utils.py

import json
import os
import shlex
import subprocess

import click
from nix_prefetch_github import nix_prefetch_github

HERE = os.path.dirname(__file__)

TO_IGNORE = ["pip", "setuptools", "wheel", "zc.buildout", "zc.recipe.egg"]

PYTHON_VERSIONS = {
    "2.6": "python26Full",
    "2.7": "python27Full",
    "3.2": "python32",
    "3.3": "python33",
    "3.4": "python34",
    "3.5": "python35",
    "3.6": "python36",
    "3.7": "python37",
    "3": "python3",
    "pypy": "pypy",
}


def pretty_option(option):
    if option is None:
        return ""
    else:
        return " [value: {}]".format(
            type(option) in [list, tuple] and " ".join(option) or option
        )


def safe(string):
    return string.replace('"', '\\"')


def cmd(command, verbose=False, stderr=subprocess.STDOUT):

    if isinstance(command, str):
        command = shlex.split(command)

    if verbose:
        click.echo("|-> " + " ".join(command))

    p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=stderr)

    out = []
    while True:
        line = p.stdout.readline().decode()
        if line == "" and p.poll() is not None:
            break
        if line != "":
            if verbose:
                click.echo("    " + line.rstrip("\n"))
            out.append(line)

    return p.returncode, "\n".join(out)


def create_command_options(options):
    command_options = []
    for name, value in options.items():
        if isinstance(value, str):
            command_options.append('--argstr {} "{}"'.format(name, value))
        elif isinstance(value, list) or isinstance(value, tuple):
            value = "[ %s ]" % (" ".join(['"%s"' % x for x in value]))
            command_options.append("--arg {} '{}'".format(name, value))
    return " ".join(command_options)


def args_as_list(inputs):
    return list(filter(lambda x: x != "", (" ".join(inputs)).split(" ")))


def prefetch_url(url):
    output = cmd(["nix-prefetch-url", url], stderr=None)
    return output[-1][:-1]


def prefetch_git(url, rev=None):
    command = ["nix-prefetch-git", url]

    if rev is not None:
        command += ["--rev", rev]
    try:
        completed_proc = subprocess.run(
            command,
            universal_newlines=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
        )
    except FileNotFoundError:
        raise click.ClickException(
            "Could not find executable `nix-prefetch-git`.  "
            "Make sure it is installed correctly and available in "
            "$PATH."
        )

    returncode = completed_proc.returncode

    if returncode != 0:
        raise click.ClickException(
            (
                "Could not fetch git repository at {url}, git returncode was "
                "{code}. stdout:\n{stdout}\nstderr:\n{stderr}"
            ).format(
                url=url,
                code=returncode,
                stdout=completed_proc.stdout,
                stderr=completed_proc.stderr,
            )
        )
    repo_data = json.loads(completed_proc.stdout)
    return repo_data


def prefetch_github(owner, repo, rev=None):
    return nix_prefetch_github(owner, repo, rev=rev)