Loading spinner

Author: rakhim

verda/cli/commands/balance.py

@@ -5,7 +5,7 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import console, output_json
+from verda.cli.utils.output import console, output_json, spinner
 
 app = typer.Typer(no_args_is_help=True)
 
@@ -15,7 +15,8 @@
 def get_balance() -> None:
     """Get account balance."""
     client = get_client()
-    balance = client.balance.get()
+    with spinner('Fetching balance...'):
+        balance = client.balance.get()
 
     if cli_main.state['json_output']:
         output_json({'amount': balance.amount, 'currency': balance.currency})

verda/cli/commands/clusters.py

@@ -7,7 +7,14 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import console, output_json, output_single, output_table, success
+from verda.cli.utils.output import (
+    console,
+    output_json,
+    output_single,
+    output_table,
+    spinner,
+    success,
+)
 from verda.constants import ClusterStatus, Locations
 
 app = typer.Typer(no_args_is_help=True)
@@ -46,7 +53,8 @@ def list_clusters(
 ) -> None:
     """List all clusters."""
     client = get_client()
-    clusters = client.clusters.get(status=status)
+    with spinner('Fetching clusters...'):
+        clusters = client.clusters.get(status=status)
 
     if cli_main.state['json_output']:
         output_json(clusters)
@@ -61,7 +69,8 @@ def get_cluster(
 ) -> None:
     """Get cluster details by ID."""
     client = get_client()
-    cluster = client.clusters.get_by_id(cluster_id)
+    with spinner('Fetching cluster...'):
+        cluster = client.clusters.get_by_id(cluster_id)
 
     if cli_main.state['json_output']:
         output_json(cluster)
@@ -131,18 +140,19 @@ def create_cluster(
 
     wait_status = None if no_wait else ClusterStatus.PROVISIONING
 
-    cluster = client.clusters.create(
-        cluster_type=cluster_type,
-        image=image,
-        hostname=hostname,
-        description=description,
-        ssh_key_ids=ssh_key_ids or [],
-        location=location,
-        startup_script_id=startup_script_id,
-        shared_volume_name=shared_volume_name,
-        shared_volume_size=shared_volume_size,
-        wait_for_status=wait_status,
-    )
+    with spinner('Creating cluster...'):
+        cluster = client.clusters.create(
+            cluster_type=cluster_type,
+            image=image,
+            hostname=hostname,
+            description=description,
+            ssh_key_ids=ssh_key_ids or [],
+            location=location,
+            startup_script_id=startup_script_id,
+            shared_volume_name=shared_volume_name,
+            shared_volume_size=shared_volume_size,
+            wait_for_status=wait_status,
+        )
 
     if cli_main.state['json_output']:
         output_json(cluster)
@@ -166,7 +176,8 @@ def delete_cluster(
             raise typer.Abort()
 
     client = get_client()
-    client.clusters.delete(cluster_id)
+    with spinner('Deleting cluster...'):
+        client.clusters.delete(cluster_id)
     success(f'Cluster {cluster_id} deletion initiated')
 
 
@@ -186,14 +197,16 @@ def check_availability(
     client = get_client()
 
     if cluster_type:
-        available = client.clusters.is_available(cluster_type, location_code=location)
+        with spinner('Checking availability...'):
+            available = client.clusters.is_available(cluster_type, location_code=location)
         if cli_main.state['json_output']:
             output_json({'cluster_type': cluster_type, 'available': available})
         else:
             status = '[green]Available[/green]' if available else '[red]Not Available[/red]'
             typer.echo(f'{cluster_type}: {status}')
     else:
-        availabilities = client.clusters.get_availabilities(location_code=location)
+        with spinner('Fetching availabilities...'):
+            availabilities = client.clusters.get_availabilities(location_code=location)
         if cli_main.state['json_output']:
             output_json(availabilities)
         else:
@@ -212,7 +225,8 @@ def list_cluster_images(
 ) -> None:
     """List available cluster images."""
     client = get_client()
-    images = client.clusters.get_cluster_images(cluster_type=cluster_type)
+    with spinner('Fetching images...'):
+        images = client.clusters.get_cluster_images(cluster_type=cluster_type)
 
     if cli_main.state['json_output']:
         output_json(images)

verda/cli/commands/images.py

@@ -5,7 +5,7 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import output_json, output_table
+from verda.cli.utils.output import output_json, output_table, spinner
 
 app = typer.Typer(no_args_is_help=True)
 
@@ -15,7 +15,8 @@
 def list_images() -> None:
     """List available OS images."""
     client = get_client()
-    images = client.images.get()
+    with spinner('Fetching images...'):
+        images = client.images.get()
 
     if cli_main.state['json_output']:
         output_json(images)

verda/cli/commands/instance_types.py

@@ -5,7 +5,7 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import output_json, output_table
+from verda.cli.utils.output import output_json, output_table, spinner
 
 app = typer.Typer(no_args_is_help=True)
 
@@ -29,7 +29,8 @@ def format_memory(memory: dict) -> str:
 def list_instance_types() -> None:
     """List available instance types with pricing."""
     client = get_client()
-    types = client.instance_types.get()
+    with spinner('Fetching instance types...'):
+        types = client.instance_types.get()
 
     if cli_main.state['json_output']:
         output_json(types)

verda/cli/commands/instances.py

@@ -7,7 +7,7 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import output_json, output_single, output_table, success
+from verda.cli.utils.output import output_json, output_single, output_table, spinner, success
 from verda.constants import Actions, Locations
 
 app = typer.Typer(no_args_is_help=True)
@@ -49,7 +49,8 @@ def list_instances(
 ) -> None:
     """List all instances."""
     client = get_client()
-    instances = client.instances.get(status=status)
+    with spinner('Fetching instances...'):
+        instances = client.instances.get(status=status)
 
     if cli_main.state['json_output']:
         output_json(instances)
@@ -64,7 +65,8 @@ def get_instance(
 ) -> None:
     """Get instance details by ID."""
     client = get_client()
-    instance = client.instances.get_by_id(instance_id)
+    with spinner('Fetching instance...'):
+        instance = client.instances.get_by_id(instance_id)
 
     if cli_main.state['json_output']:
         output_json(instance)
@@ -140,19 +142,20 @@ def create_instance(
 
     max_wait_time = 0 if no_wait else 180
 
-    instance = client.instances.create(
-        instance_type=instance_type,
-        image=image,
-        hostname=hostname,
-        description=description,
-        ssh_key_ids=ssh_key_ids or [],
-        location=location,
-        startup_script_id=startup_script_id,
-        existing_volumes=existing_volumes,
-        is_spot=spot,
-        contract=contract,
-        max_wait_time=max_wait_time,
-    )
+    with spinner('Creating instance...'):
+        instance = client.instances.create(
+            instance_type=instance_type,
+            image=image,
+            hostname=hostname,
+            description=description,
+            ssh_key_ids=ssh_key_ids or [],
+            location=location,
+            startup_script_id=startup_script_id,
+            existing_volumes=existing_volumes,
+            is_spot=spot,
+            contract=contract,
+            max_wait_time=max_wait_time,
+        )
 
     if cli_main.state['json_output']:
         output_json(instance)
@@ -175,7 +178,8 @@ def start_instance(
 ) -> None:
     """Start an instance."""
     client = get_client()
-    client.instances.action(instance_id, Actions.START)
+    with spinner('Starting instance...'):
+        client.instances.action(instance_id, Actions.START)
     success(f'Instance {instance_id} start initiated')
 
 
@@ -186,7 +190,8 @@ def stop_instance(
 ) -> None:
     """Stop (shutdown) an instance."""
     client = get_client()
-    client.instances.action(instance_id, Actions.SHUTDOWN)
+    with spinner('Stopping instance...'):
+        client.instances.action(instance_id, Actions.SHUTDOWN)
     success(f'Instance {instance_id} shutdown initiated')
 
 
@@ -206,7 +211,8 @@ def delete_instance(
             raise typer.Abort()
 
     client = get_client()
-    client.instances.action(instance_id, Actions.DELETE)
+    with spinner('Deleting instance...'):
+        client.instances.action(instance_id, Actions.DELETE)
     success(f'Instance {instance_id} deletion initiated')
 
 
@@ -217,7 +223,8 @@ def hibernate_instance(
 ) -> None:
     """Hibernate an instance."""
     client = get_client()
-    client.instances.action(instance_id, Actions.HIBERNATE)
+    with spinner('Hibernating instance...'):
+        client.instances.action(instance_id, Actions.HIBERNATE)
     success(f'Instance {instance_id} hibernation initiated')
 
 
@@ -228,7 +235,8 @@ def restore_instance(
 ) -> None:
     """Restore a hibernated instance."""
     client = get_client()
-    client.instances.action(instance_id, Actions.RESTORE)
+    with spinner('Restoring instance...'):
+        client.instances.action(instance_id, Actions.RESTORE)
     success(f'Instance {instance_id} restore initiated')
 
 
@@ -252,16 +260,20 @@ def check_availability(
     client = get_client()
 
     if instance_type:
-        available = client.instances.is_available(
-            instance_type, is_spot=spot, location_code=location
-        )
+        with spinner('Checking availability...'):
+            available = client.instances.is_available(
+                instance_type, is_spot=spot, location_code=location
+            )
         if cli_main.state['json_output']:
             output_json({'instance_type': instance_type, 'available': available})
         else:
             status = '[green]Available[/green]' if available else '[red]Not Available[/red]'
             typer.echo(f'{instance_type}: {status}')
     else:
-        availabilities = client.instances.get_availabilities(is_spot=spot, location_code=location)
+        with spinner('Fetching availabilities...'):
+            availabilities = client.instances.get_availabilities(
+                is_spot=spot, location_code=location
+            )
         if cli_main.state['json_output']:
             output_json(availabilities)
         else:

verda/cli/commands/locations.py

@@ -5,7 +5,7 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import output_json, output_table
+from verda.cli.utils.output import output_json, output_table, spinner
 
 app = typer.Typer(no_args_is_help=True)
 
@@ -15,7 +15,8 @@
 def list_locations() -> None:
     """List available datacenter locations."""
     client = get_client()
-    locations = client.locations.get()
+    with spinner('Fetching locations...'):
+        locations = client.locations.get()
 
     if cli_main.state['json_output']:
         output_json(locations)

verda/cli/commands/ssh_keys.py

@@ -8,7 +8,7 @@
 from verda.cli import main as cli_main
 from verda.cli.utils.client import get_client
 from verda.cli.utils.errors import handle_api_errors
-from verda.cli.utils.output import output_json, output_single, output_table, success
+from verda.cli.utils.output import output_json, output_single, output_table, spinner, success
 
 app = typer.Typer(no_args_is_help=True)
 
@@ -24,7 +24,8 @@
 def list_ssh_keys() -> None:
     """List all SSH keys."""
     client = get_client()
-    keys = client.ssh_keys.get()
+    with spinner('Fetching SSH keys...'):
+        keys = client.ssh_keys.get()
 
     if cli_main.state['json_output']:
         output_json(keys)
@@ -39,7 +40,8 @@ def get_ssh_key(
 ) -> None:
     """Get SSH key details by ID."""
     client = get_client()
-    key = client.ssh_keys.get_by_id(key_id)
+    with spinner('Fetching SSH key...'):
+        key = client.ssh_keys.get_by_id(key_id)
 
     if cli_main.state['json_output']:
         output_json(key)
@@ -77,7 +79,8 @@ def create_ssh_key(
         key = key_file.read_text().strip()
 
     client = get_client()
-    ssh_key = client.ssh_keys.create(name, key)
+    with spinner('Creating SSH key...'):
+        ssh_key = client.ssh_keys.create(name, key)
 
     if cli_main.state['json_output']:
         output_json(ssh_key)
@@ -101,8 +104,9 @@ def delete_ssh_key(
             raise typer.Abort()
 
     client = get_client()
-    if len(key_ids) == 1:
-        client.ssh_keys.delete_by_id(key_ids[0])
-    else:
-        client.ssh_keys.delete(key_ids)
+    with spinner('Deleting SSH key(s)...'):
+        if len(key_ids) == 1:
+            client.ssh_keys.delete_by_id(key_ids[0])
+        else:
+            client.ssh_keys.delete(key_ids)
     success('SSH key(s) deleted')