From 05b45d3ade3260526fc00c5ab53dad4ea7ed290b Mon Sep 17 00:00:00 2001
From: nhertz <nhertz@nrao.edu>
Date: Thu, 20 Aug 2020 16:08:22 -0600
Subject: [PATCH] Added username/password collection to ensure a connection to
 builder.

---
 build/tools/transfer_to_builder.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/build/tools/transfer_to_builder.py b/build/tools/transfer_to_builder.py
index 6ae2edf2a..64d7e4fc7 100644
--- a/build/tools/transfer_to_builder.py
+++ b/build/tools/transfer_to_builder.py
@@ -1,4 +1,9 @@
-import subprocess, paramiko, fnmatch, os
+import subprocess
+import paramiko
+import fnmatch
+import os
+import getpass
+
 from scp import SCPClient
 
 def get_build_pkg_names():
@@ -23,7 +28,11 @@ def create_ssh_client(server):
     client = paramiko.SSHClient()
     client.load_system_host_keys()
     client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-    client.connect(server)
+
+    username = input("Enter NRAO username: ")
+    password = getpass.getpass(prompt="Enter NRAO password: ")
+
+    client.connect(server, username=username, password=password)
     return client
 
 def transfer_packages(pkg_names):
@@ -34,9 +43,9 @@ def transfer_packages(pkg_names):
     if len(pkg_names):
         builder_addr = "builder.aoc.nrao.edu"
         builder_path = "/home/builder.aoc.nrao.edu/content/conda/noarch"
-        ssh = create_ssh_client(builder_addr)
-        with SCPClient(ssh.get_transport()) as scp:
-            [scp.put(pkg, builder_path) for pkg in pkg_names]
+        with create_ssh_client(builder_addr) as ssh:
+            with SCPClient(ssh.get_transport()) as scp:
+                [scp.put(pkg, builder_path) for pkg in pkg_names]
         cmd_cd = "cd {}".format(builder_path)
         cmd_index = "conda index .."
         cmd_chmod = "chmod -f 664 *"
-- 
GitLab