Merge install scripts

[perso/Immae/Projets/Puppet.git] / python / reboot_ovh_cloud_instance.py

diff --git a/python/reboot_ovh_cloud_instance.py b/python/reboot_ovh_cloud_instance.py
new file mode 100644 (file)
index 0000000..de20c07
--- /dev/null
+++ b/python/reboot_ovh_cloud_instance.py
@@ -0,0 +1,28 @@
+# -*- encoding: utf-8 -*-
+import json
+try:
+    from ovh import ovh
+except ImportError:
+    # In case it's installed globally
+    import ovh
+import sys
+from ovh_helper import find_cloud_instance
+
+# Credentials are stored in ~/.ovh.conf
+# See ovh/README.rst
+client = ovh.Client()
+
+project, instance = find_cloud_instance(client, sys.argv[-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)
+