Programmatic instance management

1
import requests
2
from pathlib import Path
3
import time
4
import json
5
6
base_url = 'https://platform.fluidstack.io/'
7
api_key = 'api_key_j123My_FakeAPIKey'
8
9
# Set the API key in a header
10
headers = {
11
    'api-key': api_key,
12
    'Content-Type': 'application/json'
13
}
14
15
# 1: Add an SSH key
16
print('Adding SSH key')
17
18
# Read the public key from ~/.ssh/id_rsa.pub
19
public_key = Path.home().joinpath('.ssh', 'id_rsa.pub').read_text().strip()
20
21
ssh_key_name = 'python_ssh_key'
22
23
# Send POST request to /ssh_keys
24
response = requests.post(
25
    base_url + 'ssh_keys',
26
    headers=headers,
27
    json={
28
        'name': ssh_key_name,
29
        'public_key': public_key
30
    },
31
)
32
33
# Print response body if the response status code is not OK
34
if not response.ok:
35
    raise Exception('Could not add SSH key')
36
37
# 2: Create an instance
38
print('Creating instance')
39
40
# Send POST request to /instances
41
created_instance = requests.post(
42
    base_url + 'instances',
43
    headers=headers,
44
    json={
45
        'name': 'python_test_instance',
46
        'gpu_type': 'RTX_A6000_48GB',
47
        'ssh_key': ssh_key_name,
48
    },
49
)
50
51
print('Response: ' + json.dumps(created_instance.json(), indent=2))
52
if not created_instance.ok:
53
    raise Exception('Could not create instance')
54
55
instance_id = created_instance.json()['id']
56
57
status = 'pending'
58
print('Waiting for instance to start')
59
60
# 3: Check the instance status using GET /instances/{instance_id}
61
# Check status every 10 seconds, exit when status is 'running'
62
count = 1
63
while status != 'running':
64
    if status != 'pending':
65
        print('Could not create instance')
66
        raise Exception('Could not create instance')
67
    try:
68
        # Send GET request to /instances/{instance_id}
69
        response = requests.get(base_url + f'instances/{instance_id}', headers=headers)
70
        instance = response.json()
71
        status = instance['status']
72
    except Exception as e:
73
        print(e)
74
    print(f'\t [{count}] Status: {status}')
75
    time.sleep(10)
76
    count += 1
77
78
# 4: Print the instance IP
79
print('Instance IP address: ' + instance['ip_address'])
80
81
# 5: Terminate the instance
82
print('Terminating instance')
83
84
# Send DELETE request to /instances/{instance_id}
85
response = requests.delete(base_url + f'instances/{instance_id}', headers=headers)
86
if not response.ok:
87
    print('Response: ' + json.dumps(response.json(), indent=2))
88
    raise Exception('Could not delete instance')
89
print('Instance deleted')
90
91
# 6: Delete the public SSH key that was added to the account
92
response = requests.delete(base_url + f'ssh_keys/{ssh_key_name}', headers=headers)
93
if not response.ok:
94
    print('Response: ' + json.dumps(response.json(), indent=2))
95
    raise Exception('Could not delete the SSH key')
96
print('SSH key deleted')