diff --git a/src/subscription/azext_subscription/_params.py b/src/subscription/azext_subscription/_params.py index 4b011ac3f71..d3da360fef6 100644 --- a/src/subscription/azext_subscription/_params.py +++ b/src/subscription/azext_subscription/_params.py @@ -4,13 +4,14 @@ # -------------------------------------------------------------------------------------------- from azure.cli.core.commands.parameters import get_enum_type +from .subscription.models import OfferType # pylint: disable=line-too-long def load_arguments(self, _): with self.argument_context('account create') as c: c.argument('enrollment_account_name', options_list=('--enrollment-account-name', '--enrollment-account-object-id'), required=True, help='The name of the enrollment account which should be used to create a subscription.') - c.argument('offer_type', required=True, help='The offer type of the subscription. For example, MS-AZR-0017P (EnterpriseAgreement) and MS-AZR-0148P (EnterpriseAgreement devTest) are available.', arg_type=get_enum_type(['MS-AZR-0017P', 'MS-AZR-0148P'])) + c.argument('offer_type', required=True, help='The offer type of the subscription. For example, MS-AZR-0017P (EnterpriseAgreement) and MS-AZR-0148P (EnterpriseAgreement devTest) are available.', arg_type=get_enum_type(OfferType)) c.argument('display_name', help='The display name of the subscription.') c.argument('owner_object_id', help='The object id(s) of the owner(s) which should be granted access to the new subscription.') c.argument('owner_spn', help='The service principal name(s) of the owner(s) which should be granted access to the new subscription.') diff --git a/src/subscription/azext_subscription/subscription/models/subscription_client_enums.py b/src/subscription/azext_subscription/subscription/models/subscription_client_enums.py index 95601304f53..bee6b315249 100644 --- a/src/subscription/azext_subscription/subscription/models/subscription_client_enums.py +++ b/src/subscription/azext_subscription/subscription/models/subscription_client_enums.py @@ -16,6 +16,9 @@ class OfferType(str, Enum): ms_azr_0017_p = "MS-AZR-0017P" ms_azr_0148_p = "MS-AZR-0148P" + ms_azr_usgov_0017p = "MS-AZR-USGOV-0017P" + ms_azr_usgov_0015p = "MS-AZR-USGOV-0015P" + ms_azr_usgov_0148p = "MS-AZR-USGOV-0148P" class SubscriptionState(str, Enum):