kk6/snake-pit

# -*- coding: utf-8 -*-
"""Main `pit` CLI."""
from __future__ import absolute_import, division, print_function, unicode_literals

import os
import sys
from textwrap import dedent

import click

from . import __version__
from . import echoes
from .adapters import pip
from .config import DEFAULT_CONFIG, get_config, get_requirements_file
from .dists import DistFinder
from .exceptions import (
    ConfigDoesNotExist,
    InvalidConfiguration,
    RequirementsKeyError,
)
from .groups import AliasedGroup
from .utils import re_edit

CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'])
CONFIG_PATH = os.environ.get('PIT_CONFIG_PATH', 'pit.yml')
WHITE_LIST = ('pip', 'setuptools', 'click', 'distlib', 'pyyaml')


@click.group(cls=AliasedGroup, context_settings=CONTEXT_SETTINGS)
@click.version_option(__version__, '--version', '-V', prog_name='snake-pit')
@click.pass_context
def cli(ctx):
    """Depending on management packages, and then edit the requirements file.
    """
    ctx.obj = {}
    try:
        ctx.obj['CONFIG'] = get_config(CONFIG_PATH)
    except (ConfigDoesNotExist, InvalidConfiguration) as e:
        echoes.err(str(e))
        echoes.warn("Instead, it uses the default configuration.")
        ctx.obj['CONFIG'] = DEFAULT_CONFIG


@cli.command()
@click.argument('packages', nargs=-1, required=True)
@click.option(
    '--requirement', '-r', type=click.Path(exists=True),
    help="Append package names to the given requirement file."
)
@click.option('--quiet', '-q', is_flag=True, help='Do not display the output result.')
@click.option('--name', '-n', help=dedent("""\
    Specify requirements file name.
    This option takes precedence over the '-r' option.
    """))
@click.option(
    '--pre', is_flag=True,
    help=("Include pre-release and development versions. "
          "By default, pip only finds stable versions.")
)
@click.pass_context
def install(ctx, packages, requirement, quiet, name, pre):
    """Install packages and write requirements file.

    To append package names were installed in requirements file,
    if the 'pip install' was successful.

    """
    try:
        requirements_file = get_requirements_file(
            ctx.obj['CONFIG'], requirement, name, mode='a+'
        )
    except RequirementsKeyError as e:
        echoes.err(str(e))
        sys.exit(2)

    finder = DistFinder()

    upgradable_packages = finder.choose_installed(packages)
    if upgradable_packages:
        echoes.info(
            "Following packages installed. "
            "(use pip install --upgrade to upgrade): {}".format(
                ", ".join(upgradable_packages)
            )
        )

    installable_packages = finder.choose_not_installed(packages)
    if not installable_packages:
        echoes.warn("There is no installable packages a new.")
        sys.exit(0)

    pip_options = []
    if pre:
        pip_options.append('--pre')
    raised = pip.install(installable_packages, pip_options)
    if raised:
        sys.exit(2)

    requirements_file.write('\n'.join(packages) + '\n')
    echoes.info("Append the following packages in {requirement}: {packages}".format(
        requirement=requirements_file.name, packages=", ".join(installable_packages)
    ))

    if not quiet:
        requirements_file.seek(0)
        echoes.info("{} has been updated as follows:".format(requirements_file.name))
        echoes.success(requirements_file.read())
        requirements_file.close()


@cli.command()
@click.argument('packages', nargs=-1, required=True,
                callback=lambda ctx, param, val: [v.lower() for v in val])
@click.option(
    '--requirement', '-r', type=click.Path(exists=True),
    help="Remove package names from the given requirement file."
)
@click.option('--quiet', '-q', is_flag=True, help='Do not display the output result.')
@click.option('--name', '-n', help=dedent("""\
    Specify requirements file name.
    This option takes precedence over the '-r' option.
    """))
@click.option('--auto', '-a', is_flag=True,
              help='Uninstall as much as possible the dependent packages.')
@click.pass_context
def uninstall(ctx, packages, requirement, quiet, name, auto):
    """Uninstall packages and remove from requirements file.

    To Remove uninstalled packages from requirements file,
    if the 'pip uninstall' was successful.

    """
    try:
        requirements_file = get_requirements_file(
            ctx.obj['CONFIG'], requirement, name, mode='r'
        )
    except RequirementsKeyError as e:
        echoes.err(str(e))
        sys.exit(2)

    finder = DistFinder(WHITE_LIST)

    # Warning packages that are not installed.
    not_installed = finder.choose_not_installed(packages)
    if not_installed:
        echoes.err(
            "Following packages are not installed: {}".format(
                ", ".join(not_installed)
            )
        )

    # Run the uninstallation.
    installed = finder.choose_installed(packages)
    if not installed:
        echoes.err("There is no uninstall possible package.")
        sys.exit(2)

    deletable_set = set(installed)
    if auto:
        for pkg in installed:
            deletable = finder.get_deletable_dist_set(pkg)
            deletable_set |= deletable
        echoes.info(
            "Specified package and becomes unnecessary by which they are removed, "
            "it will remove the following packages:\n\n{}\n".format("\n".join(deletable_set))
        )
        click.confirm("Are you sure?", abort=True)

    if pip.uninstall(deletable_set):
        sys.exit(2)

    content = re_edit(requirements_file.readlines(), packages)
    with open(requirements_file.name, 'w') as f:
        f.write(content)

    echoes.info("Remove the following packages from {requirement}: {packages}".format(
        requirement=requirements_file.name, packages=", ".join(deletable_set)
    ))

    if not quiet:
        echoes.info("{} has been updated as follows:".format(requirements_file.name))
        echoes.success(content)