version_check.py

# -*- coding: utf-8 -*-
"""Version-check script."""
import os
import sys
import codecs
Failed = 0
VERSION = "0.50"


SETUP_ITEMS = [
    "version='{0}'",
    'https://github.com/Moduland/Orangetool/tarball/v{0}']
README_ITEMS = [
    "[Version {0}](https://github.com/moduland/Orangetool/archive/v{0}.zip)",
    "pip install orangetool=={0}",
    "pip3 install orangetool=={0}"]
CHANGELOG_ITEMS = [
    "## [{0}]",
    "https://github.com/Moduland/Orangetool/compare/v{0}...dev",
    "[{0}]:"]
PARAMS_ITEMS = ['ORANGETOOL_VERSION = "{0}"']
FILES = {
    "setup.py": SETUP_ITEMS,
    "README.md": README_ITEMS,
    "CHANGELOG.md": CHANGELOG_ITEMS,
    os.path.join(
        "orangetool",
        "orangetool_params.py"): PARAMS_ITEMS}

TEST_NUMBER = len(FILES.keys())


def print_result(failed=False):
    """
    Print final result.

    :param failed: failed flag
    :type failed: bool
    :return: None
    """
    message = "Version tag tests "
    if not failed:
        print("\n" + message + "passed!")
    else:
        print("\n" + message + "failed!")
    print("Passed : " + str(TEST_NUMBER - Failed) + "/" + str(TEST_NUMBER))


if __name__ == "__main__":
    for file_name in FILES.keys():
        try:
            file_content = codecs.open(
                file_name, "r", "utf-8", 'ignore').read()
            for test_item in FILES[file_name]:
                if file_content.find(test_item.format(VERSION)) == -1:
                    print("Incorrect version tag in " + file_name)
                    Failed += 1
                    break
        except Exception as e:
            Failed += 1
            print("Error in " + file_name + "\n" + "Message : " + str(e))

    if Failed == 0:
        print_result(False)
        sys.exit(0)
    else:
        print_result(True)
        sys.exit(1)