a

app/ceph.py

@@ -11,14 +11,13 @@ def ceph():
 @click.option("--ceph_url", required=True)
 @click.option("--ssh_username", required=True)
 def setup(ceph_url, ssh_username):
-    ssh_key = ""
-    with open(os.path.expanduser("~/.ssh/id_rsa.pub")) as pubkey:
-        ssh_key = pubkey.read()
-    
-    with open(os.path.expanduser("~/.ssh/known_hosts"), "a") as known_hosts:
-        known_hosts.write(f"{ceph_url} {ssh_key}")
-
-    os.makedirs("/etc/ceph", exist_ok=True)
-    command = f"sftp -b ./ceph_batch_cmd {ssh_username}@{ceph_url}:/etc/ceph"
-    subprocess.check_output(command.split())
-    
+    try:
+        os.makedirs("/etc/ceph", exist_ok=True)
+        command = f"sftp -b ./ceph_batch_cmd {ssh_username}@{ceph_url}:/etc/ceph"
+        subprocess.check_output(command.split())
+    except Exception:
+        p = subprocess.check_output(f"ssh-keyscan {ceph_url}".split())
+        keys = p.decode("utf-8").strip().split("\n")
+        with open(os.path.expanduser("~/.ssh/known_hosts", "a")) as known_hosts:
+            for key in keys:
+                known_hosts.write(key)