path: root/python/reboot_cloud_instance.py

# -*- encoding: utf-8 -*-
import json
try:
    from ovh import ovh
except ImportError:
    # In case it's installed globally
    import ovh
import sys

# Credentials are stored in ~/.ovh.conf
# See ovh/README.rst
client = ovh.Client()

projects_list = client.get('/cloud/project/')
if len(projects_list) > 1:
    print("More than one project is not supported, taking the first one")
project = projects_list[0]
instances_list = client.get('/cloud/project/{}/instance'.format(project))
instances = dict(map(lambda x: (x["id"], x), instances_list))
if sys.argv[-1] in instances:
    instance = instances[sys.argv[-1]]
else:
    print("Instance not in list:")
    for instance in instances_list:
        print("{}: {}".format(instance["name"], instance["id"]))
    sys.exit(1)

if "--rescue" in sys.argv:
    netboot_mode="rescue"
elif "--local" in sys.argv:
    netboot_mode="local"
else:
    netboot_mode=None

if netboot_mode is not None:
    result = client.post("/cloud/project/{}/instance/{}/rescueMode".format(project,
        instance["id"]), imageId=instance["imageId"], rescue=(netboot_mode == "rescue"))
    print(result)