From 260c9dc10800d8d0462fd3098b5d4c354b327d75 Mon Sep 17 00:00:00 2001 From: Derek Bekoe Date: Fri, 5 Aug 2016 10:55:30 -0700 Subject: [PATCH] Fix: Check for AttributeError as location is not always specified. --- .../azure-cli-vm/azure/cli/command_modules/vm/_params.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/command_modules/azure-cli-vm/azure/cli/command_modules/vm/_params.py b/src/command_modules/azure-cli-vm/azure/cli/command_modules/vm/_params.py index 37c5842c2c5..9e2b2edb59c 100644 --- a/src/command_modules/azure-cli-vm/azure/cli/command_modules/vm/_params.py +++ b/src/command_modules/azure-cli-vm/azure/cli/command_modules/vm/_params.py @@ -41,7 +41,10 @@ def get_urn_aliases_completion_list(prefix, **kwargs):#pylint: disable=unused-ar return [i['urn alias'] for i in images] def get_vm_size_completion_list(prefix, action, parsed_args, **kwargs):#pylint: disable=unused-argument - location = parsed_args.location if parsed_args.location else get_one_of_subscription_locations() + try: + location = parsed_args.location + except AttributeError: + location = get_one_of_subscription_locations() result = get_vm_sizes(location) return [r.name for r in result]