pyinfra 0.11.dev3__py3-none-any.whl → 3.5.1__py3-none-any.whl

pyinfra/api/connectors/docker.py

@@ -1,178 +0,0 @@
-import os
-
-from tempfile import mkstemp
-
-import click
-
-import pyinfra
-
-from pyinfra import local, logger
-from pyinfra.api.exceptions import InventoryError
-from pyinfra.api.util import get_file_io, memoize
-from pyinfra.progress import progress_spinner
-
-from .local import run_shell_command as run_local_shell_command
-
-
-@memoize
-def show_warning():
-    logger.warning('The @docker connector is in Alpha!')
-
-
-def make_names_data(image=None):
-    if not image:
-        raise InventoryError('No docker base image provided!')
-
-    show_warning()
-
-    # Save the image as the hostname, no data, @docker group
-    yield image, {}, ['@docker']
-
-
-def connect(state, host, for_fact=None):
-    if 'docker_container_id' in host.host_data:  # user can provide a docker_container_id
-        return True
-
-    with progress_spinner({'docker run'}):
-        container_id = local.shell(
-            'docker run -d {0} sleep 10000'.format(host.name),
-            splitlines=True,
-        )[-1]  # last line is the container ID
-
-    host.host_data['docker_container_id'] = container_id
-    return True
-
-
-def disconnect(state, host):
-    if not pyinfra.is_cli:
-        return
-
-    container_id = host.host_data['docker_container_id'][:12]
-
-    with progress_spinner({'docker commit'}):
-        image_id = local.shell(
-            'docker commit {0}'.format(container_id),
-            splitlines=True,
-        )[-1][7:19]  # last line is the image ID, get sha256:[XXXXXXXXXX]...
-
-    with progress_spinner({'docker rm'}):
-        local.shell(
-            'docker rm -f {0}'.format(container_id),
-        )
-
-    logger.info('{0}docker build complete, image ID: {1}'.format(
-        host.print_prefix, click.style(image_id, bold=True),
-    ))
-
-
-def run_shell_command(
-    state, host, command,
-    timeout=None, print_output=False,
-    return_combined_output=False,
-    **kwargs  # ignored (sudo/etc)
-):
-    container_id = host.host_data['docker_container_id']
-    docker_command = 'docker exec {0} sh -c "{1}"'.format(container_id, command)
-
-    return run_local_shell_command(
-        state, host, docker_command,
-        timeout=timeout,
-        print_output=print_output,
-        return_combined_output=return_combined_output,
-    )
-
-
-def put_file(
-    state, host, filename_or_io, remote_filename,
-    print_output=False,
-    **kwargs  # ignored (sudo/etc)
-):
-    '''
-    Upload a file/IO object to the target Docker container by copying it to a
-    temporary location and then uploading it into the container using ``docker cp``.
-    '''
-
-    _, temp_filename = mkstemp()
-
-    try:
-        # Load our file or IO object and write it to the temporary file
-        with get_file_io(filename_or_io) as file_io:
-            with open(temp_filename, 'wb') as temp_f:
-                data = file_io.read()
-
-                if isinstance(data, str):
-                    data = data.encode()
-
-                temp_f.write(data)
-
-        docker_id = host.host_data['docker_container_id']
-        docker_command = 'docker cp {0} {1}:{2}'.format(
-            temp_filename,
-            docker_id,
-            remote_filename,
-        )
-
-        status, _, stderr = run_local_shell_command(
-            state, host, docker_command,
-            print_output=print_output,
-        )
-    finally:
-        os.remove(temp_filename)
-
-    if not status:
-        raise IOError('\n'.join(stderr))
-
-    if print_output:
-        print('{0}file uploaded to container: {1}'.format(
-            host.print_prefix, remote_filename,
-        ))
-
-    return status
-
-
-def get_file(
-    state, host, remote_filename, filename_or_io,
-    print_output=False,
-    **kwargs  # ignored (sudo/etc)
-):
-    '''
-    Download a file from the target Docker container by copying it to a temporary
-    location and then reading that into our final file/IO object.
-    '''
-
-    _, temp_filename = mkstemp()
-
-    try:
-        docker_id = host.host_data['docker_container_id']
-        docker_command = 'docker cp {0}:{1} {2}'.format(
-            docker_id,
-            remote_filename,
-            temp_filename,
-        )
-
-        status, _, stderr = run_local_shell_command(
-            state, host, docker_command,
-            print_output=print_output,
-        )
-
-        # Load the temporary file and write it to our file or IO object
-        with open(temp_filename) as temp_f:
-            with get_file_io(filename_or_io, 'wb') as file_io:
-                data = temp_f.read()
-
-                if isinstance(data, str):
-                    data = data.encode()
-
-                file_io.write(data)
-    finally:
-        os.remove(temp_filename)
-
-    if not status:
-        raise IOError('\n'.join(stderr))
-
-    if print_output:
-        print('{0}file downloaded from container: {1}'.format(
-            host.print_prefix, remote_filename,
-        ))
-
-    return status

pyinfra/api/connectors/local.py

@@ -1,169 +0,0 @@
-import os
-
-from subprocess import PIPE, Popen
-from tempfile import mkstemp
-
-import click
-
-from pyinfra import logger
-from pyinfra.api.util import get_file_io, make_command
-
-from .util import read_buffers_into_queue, split_combined_output
-
-
-def connect(state, host, for_fact=None):
-    # Log
-    log_message = '{0}{1}'.format(
-        host.print_prefix,
-        click.style('Connected', 'green'),
-    )
-
-    if for_fact:
-        log_message = '{0}{1}'.format(
-            log_message,
-            ' (for {0} fact)'.format(for_fact),
-        )
-
-    logger.info(log_message)
-
-    return True
-
-
-def run_shell_command(
-    state, host, command,
-    get_pty=False, timeout=None, print_output=False,
-    return_combined_output=False,
-    **command_kwargs
-):
-    '''
-    Execute a command on the local machine.
-
-    Args:
-        state (``pyinfra.api.State`` obj): state object for this command
-        hostname (string): hostname of the target
-        command (string): actual command to execute
-        sudo (boolean): whether to wrap the command with sudo
-        sudo_user (string): user to sudo to
-        get_pty (boolean): whether to get a PTY before executing the command
-        env (dict): envrionment variables to set
-        timeout (int): timeout for this command to complete before erroring
-
-    Returns:
-        tuple: (exit_code, stdout, stderr)
-        stdout and stderr are both lists of strings from each buffer.
-    '''
-
-    command = make_command(command, **command_kwargs)
-
-    logger.debug('--> Running command on localhost: {0}'.format(command))
-
-    if print_output:
-        print('{0}>>> {1}'.format(host.print_prefix, command))
-
-    process = Popen(command, shell=True, stdout=PIPE, stderr=PIPE)
-
-    combined_output = read_buffers_into_queue(
-        host,
-        process.stdout,
-        process.stderr,
-        timeout=timeout,
-        print_output=print_output,
-    )
-
-    logger.debug('--> Waiting for exit status...')
-    process.wait()
-    logger.debug('--> Command exit status: {0}'.format(process.returncode))
-
-    # Close any open file descriptors
-    process.stdout.close()
-    process.stderr.close()
-
-    status = process.returncode == 0
-
-    if return_combined_output:
-        return status, combined_output
-
-    stdout, stderr = split_combined_output(combined_output)
-    return status, stdout, stderr
-
-
-def put_file(
-    state, host, filename_or_io, remote_filename,
-    print_output=False,
-    **command_kwargs
-):
-    '''
-    Upload a local file or IO object by copying it to a temporary directory
-    and then writing it to the upload location.
-    '''
-
-    _, temp_filename = mkstemp()
-
-    try:
-        # Load our file or IO object and write it to the temporary file
-        with get_file_io(filename_or_io) as file_io:
-            with open(temp_filename, 'wb') as temp_f:
-                data = file_io.read()
-
-                if isinstance(data, str):
-                    data = data.encode()
-
-                temp_f.write(data)
-
-        # Copy the file using `cp` such that we support sudo/su
-        status, _, stderr = run_shell_command(
-            state, host, 'cp {0} {1}'.format(temp_filename, remote_filename),
-            print_output=print_output,
-            **command_kwargs
-        )
-
-        if not status:
-            raise IOError('\n'.join(stderr))
-    finally:
-        os.remove(temp_filename)
-
-    if print_output:
-        print('{0}file copied: {1}'.format(host.print_prefix, remote_filename))
-
-    return status
-
-
-def get_file(
-    state, host, remote_filename, filename_or_io,
-    print_output=False,
-    **command_kwargs
-):
-    '''
-    Download a local file by copying it to a temporary location and then writing
-    it to our filename or IO object.
-    '''
-
-    _, temp_filename = mkstemp()
-
-    try:
-        # Copy the file using `cp` such that we support sudo/su
-        status, _, stderr = run_shell_command(
-            state, host, 'cp {0} {1}'.format(remote_filename, temp_filename),
-            print_output=print_output,
-            **command_kwargs
-        )
-
-        if not status:
-            raise IOError('\n'.join(stderr))
-
-        # Load our file or IO object and write it to the temporary file
-        with open(temp_filename) as temp_f:
-            with get_file_io(filename_or_io, 'wb') as file_io:
-                data = temp_f.read()
-
-                if isinstance(data, str):
-                    data = data.encode()
-
-                file_io.write(data)
-    finally:
-        os.remove(temp_filename)
-
-    if print_output:
-        print('{0}file copied: {1}'.format(host.print_prefix, remote_filename))
-
-    return True