Added price to sky status (skypilot-org#561)

* Added price to sky status

* put region and hourly price behind -a in sky status

* removed whitespace

* cache cluster region

* some touches + added computation to constructor

* forgot one fix

* formatting

Author: mraheja

sky/backends/cloud_vm_ray_backend.py

@@ -621,10 +621,11 @@ def _yield_region_zones(self, to_provision: 'resources_lib.Resources',
 
                 if prev_resources is not None and cloud.is_same_cloud(
                         prev_resources.cloud):
-                    if type(cloud) in (clouds.AWS, clouds.GCP):
+                    if cloud.is_same_cloud(sky.GCP()) or cloud.is_same_cloud(
+                            sky.AWS()):
                         region = config['provider']['region']
                         zones = config['provider']['availability_zone']
-                    elif isinstance(cloud, clouds.Azure):
+                    elif cloud.is_same_cloud(sky.Azure()):
                         region = config['provider']['location']
                         zones = None
                     else:
@@ -1135,6 +1136,7 @@ def __init__(
             self.launched_resources = launched_resources
             self.tpu_create_script = tpu_create_script
             self.tpu_delete_script = tpu_delete_script
+            self._find_cluster_region()
 
         def __repr__(self):
             return (f'ResourceHandle('
@@ -1150,6 +1152,21 @@ def __repr__(self):
         def get_cluster_name(self):
             return self.cluster_name
 
+        def _find_cluster_region(self):
+            config = backend_utils.read_yaml(self.cluster_yaml)
+            provider = config['provider']
+            cloud = self.launched_resources.cloud
+            if cloud.is_same_cloud(sky.Azure()):
+                self.cluster_region = provider['location']
+            elif cloud.is_same_cloud(sky.GCP()) or cloud.is_same_cloud(
+                    sky.AWS()):
+                self.cluster_region = provider['region']
+
+        def get_cluster_region(self):
+            if not hasattr(self, 'cluster_region'):
+                self._find_cluster_region()
+            return self.cluster_region
+
     def __init__(self):
         self.run_timestamp = backend_utils.get_run_timestamp()
         self.log_dir = os.path.join(SKY_LOGS_DIRECTORY, self.run_timestamp)

sky/cli.py

@@ -828,14 +828,22 @@ def status(all: bool, refresh: bool):  # pylint: disable=redefined-builtin
     # TODO(zhwu): Update the information for auto-stop clusters.
     show_all = all
     clusters_status = backend_utils.get_clusters(refresh)
-    cluster_table = util_lib.create_table([
+    columns = [
         'NAME',
         'LAUNCHED',
         'RESOURCES',
         'STATUS',
         'AUTOSTOP',
         'COMMAND',
-    ])
+    ]
+
+    if all:
+        columns.extend([
+            'HOURLY_PRICE',
+            'REGION',
+        ])
+
+    cluster_table = util_lib.create_table(columns)
 
     for cluster_status in clusters_status:
         launched_at = cluster_status['launched_at']
@@ -858,7 +866,7 @@ def status(all: bool, refresh: bool):  # pylint: disable=redefined-builtin
         if cluster_status['autostop'] >= 0:
             # TODO(zhwu): check the status of the autostop cluster.
             autostop_str = str(cluster_status['autostop']) + ' min'
-        cluster_table.add_row([
+        row = [
             # NAME
             cluster_status['name'],
             # LAUNCHED
@@ -872,7 +880,19 @@ def status(all: bool, refresh: bool):  # pylint: disable=redefined-builtin
             # COMMAND
             cluster_status['last_use']
             if show_all else _truncate_long_string(cluster_status['last_use']),
-        ])
+        ]
+        if all:
+            hourly_cost = handle.launched_resources.get_cost(3600) \
+                * handle.launched_nodes
+            price_str = f'$ {hourly_cost:.3f}'
+            region = handle.get_cluster_region()
+            row.extend([
+                # HOURLY PRICE
+                price_str,
+                # REGION
+                region,
+            ])
+        cluster_table.add_row(row)
     if clusters_status:
         click.echo(cluster_table)
     else: