Refactor azure.py to use subprocess_utils and ux_utils for error hand…

…ling and output handling

sky/adaptors/azure.py

@@ -7,7 +7,8 @@
 import threading
 
 from sky.adaptors import common
-from sky.utils.subprocess_utils import run
+from sky.utils import subprocess_utils
+from sky.utils import ux_utils
 
 azure = common.LazyImport(
  'azure',
@@ -22,25 +23,25 @@
 
 
 def _get_account():
- result = run('az account show -o json',
- stdout=subprocess.PIPE,
- stderr=subprocess.PIPE)
+ result = subprocess_utils.run('az account show -o json',
+  stdout=subprocess.PIPE,
+  stderr=subprocess.PIPE)
 
  if result.returncode != 0:
  error_message = result.stderr.decode()
- print(f'Error executing command: {error_message}')
- raise RuntimeError(
- 'Failed to execute az account show -o json command.')
+ with ux_utils.print_exception_no_traceback():
+ raise RuntimeError(
+ 'Failed to execute az account show -o json command. '
+ f'Error: {error_message}')
 
  try:
  return json.loads(result.stdout.decode())
  except json.JSONDecodeError as e:
  error_message = result.stderr.decode()
- print(f'JSON parsing error: {e.msg}')
- raise RuntimeError(
- f'Failed to parse JSON output. Error: {e.msg}\n'
- f'Command Error: {error_message}'
- ) from e
+ with ux_utils.print_exception_no_traceback():
+ raise RuntimeError(
+ 'Failed to parse the output of az account show -o json '
+ f'command. Error: {error_message}') from e
 
 
 def get_subscription_id() -> str: