bumpver/src/pycalver/__main__.py

#!/usr/bin/env python
# This file is part of the pycalver project
# https://gitlab.com/mbarkhau/pycalver
#
# Copyright (c) 2018 Manuel Barkhau (@mbarkhau) - MIT License
# SPDX-License-Identifier: MIT

import io
import os
import sys
import click
import logging

from . import DEBUG
from . import vcs
from . import parse
from . import config
from . import version
from . import rewrite


log = logging.getLogger("pycalver.__main__")


def _init_loggers(verbose: bool) -> None:
    if DEBUG:
        log_formatter = logging.Formatter('%(levelname)s - %(name)s - %(message)s')
        log_level     = logging.DEBUG
    elif verbose:
        log_formatter = logging.Formatter('%(levelname)s - %(message)s')
        log_level     = logging.INFO
    else:
        log_formatter = logging.Formatter('%(message)s')
        log_level     = logging.WARNING

    loggers = [log, vcs.log, parse.log, config.log, rewrite.log, version.log]

    for logger in loggers:
        if len(logger.handlers) == 0:
            ch = logging.StreamHandler(sys.stderr)
            ch.setFormatter(log_formatter)
            logger.addHandler(ch)
            logger.setLevel(log_level)

    log.debug("Loggers initialized.")


@click.group()
def cli():
    """parse and update project versions automatically."""


@cli.command()
def show() -> None:
    _init_loggers(verbose=False)

    cfg: config.MaybeConfig = config.parse()
    if cfg is None:
        log.error("Could not parse configuration from setup.cfg")
        sys.exit(1)

    print(f"Current Version: {cfg.current_version}")
    print(f"PEP440 Version: {cfg.pep440_version}")


@cli.command()
@click.argument("old_version")
@click.option(
    "--release", default=None, metavar="<name>", help="Override release name of current_version"
)
def incr(old_version: str, release: str = None) -> None:
    _init_loggers(verbose=False)

    if release and release not in parse.VALID_RELESE_VALUES:
        log.error(f"Invalid argument --release={release}")
        log.error(f"Valid arguments are: {', '.join(parse.VALID_RELESE_VALUES)}")
        sys.exit(1)

    new_version     = version.bump(old_version, release=release)
    new_version_nfo = parse.parse_version_info(new_version)

    print("PyCalVer Version:", new_version)
    print("PEP440 Version:"  , new_version_nfo.pep440_version)


@cli.command()
@click.option(
    "--dry", default=False, is_flag=True, help="Display diff of changes, don't rewrite files."
)
def init(dry: bool) -> None:
    """Initialize [pycalver] configuration in setup.cfg"""
    _init_loggers(verbose=False)

    cfg   : config.MaybeConfig = config.parse()
    if cfg:
        log.error("Configuration already initialized in setup.cfg")
        sys.exit(1)

    cfg_lines = config.default_config_lines()

    if dry:
        print("Exiting because of '--dry'. Would have written to setup.cfg:")
        print("\n    " + "\n    ".join(cfg_lines))
        return

    if os.path.exists("setup.cfg"):
        cfg_content = "\n" + "\n".join(cfg_lines)
        with io.open("setup.cfg", mode="at", encoding="utf-8") as fh:
            fh.write(cfg_content)
        print("Updated setup.cfg")
    else:
        cfg_content = "\n".join(cfg_lines)
        with io.open("setup.cfg", mode="at", encoding="utf-8") as fh:
            fh.write(cfg_content)
        print("Created setup.cfg")


@cli.command()
@click.option(
    "--release", default=None, metavar="<name>", help="Override release name of current_version"
)
@click.option("--verbose", default=False, is_flag=True, help="Log applied changes to stderr")
@click.option(
    "--dry", default=False, is_flag=True, help="Display diff of changes, don't rewrite files."
)
@click.option("--commit", default=True, is_flag=True, help="Commit after updating version strings.")
@click.option("--tag"   , default=True, is_flag=True, help="Tag the commit.")
@click.option(
    "--allow-dirty",
    default=False,
    is_flag=True,
    help=(
        "Commit even when working directory is has uncomitted changes. "
        "(WARNING: The commit will still be aborted if there are uncomitted "
        "to files with version strings."
    ),
)
def bump(
    release: str, verbose: bool, dry: bool, commit: bool, tag: bool, allow_dirty: bool
) -> None:
    _init_loggers(verbose)

    if release and release not in parse.VALID_RELESE_VALUES:
        log.error(f"Invalid argument --release={release}")
        log.error(f"Valid arguments are: {', '.join(parse.VALID_RELESE_VALUES)}")
        sys.exit(1)

    cfg: config.MaybeConfig = config.parse()

    if cfg is None:
        log.error("Could not parse configuration from setup.cfg")
        sys.exit(1)

    old_version = cfg.current_version
    new_version = version.bump(old_version, release=release)

    log.info(f"Old Version: {old_version}")
    log.info(f"New Version: {new_version}")

    if dry:
        log.info("Running with '--dry', showing diffs instead of updating files.")

    file_patterns = cfg.file_patterns
    filepaths     = set(file_patterns.keys())

    _vcs = vcs.get_vcs()
    if _vcs is None:
        log.warn("Version Control System not found, aborting commit.")
    else:
        _vcs.assert_not_dirty(filepaths, allow_dirty)

    rewrite.rewrite(new_version, file_patterns, dry, verbose)

    if dry or not commit or _vcs is None:
        return

    # TODO (mb 2018-09-04): add files and commit