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