certificate_provision/copy_cert.py

#!/usr/bin/python
import paramiko
import argparse
import os
from subprocess import list2cmdline


def connection_from_config(name):
    """Create a connection from a Host entry in the .ssh/config file"""
    config = paramiko.config.SSHConfig()
    cp = os.path.expanduser('~/.ssh/config')
    with open(cp) as file:
        config.parse(file)
    info = config.lookup(name)
    info['username'] = info.pop('user')
    info['key_filename'] = info.pop('identityfile')
    info['port'] = int(info['port'])
    client = paramiko.client.SSHClient()
    client.load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))
    client.connect(**info)
    return client


def scp(connection: paramiko.sftp_client.SFTPClient, src, dst):
    """Function to copy files from src to dst using connection"""
    connection.put(src, dst)


def exec_remote(connection: paramiko.SSHClient, command):
    """Function to execute specified command on conection"""
    if not isinstance(command, str):
        command = list2cmdline(command)
    stdin, stdout, stderr = connection.exec_command(command)
    return stdout, stderr


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        'cert_path', help="location of the certificate to install")
    parser.add_argument(
        'key_path', help="localtion of the private key to install")
    parser.add_argument(
        'store_path',
        help="directory to store the private key and certificate")
    parser.add_argument('-d', '--dry', action='store_true')
    args = parser.parse_args()