diff --git a/build/tools/transfer_to_builder.py b/build/tools/transfer_to_builder.py index 6ae2edf2a24991e2aad023b7b65237f956d132eb..64d7e4fc7ff61c4fdc8b32d4f8c7fd627f7981d8 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 *"