setup.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
@author  : Hu Ji
@file    : setup.py
@time    : 2018/11/25
@site    :
@software: PyCharm

              ,----------------,              ,---------,
         ,-----------------------,          ,"        ,"|
       ,"                      ,"|        ,"        ,"  |
      +-----------------------+  |      ,"        ,"    |
      |  .-----------------.  |  |     +---------+      |
      |  |                 |  |  |     | -==----'|      |
      |  | $ sudo rm -rf / |  |  |     |         |      |
      |  |                 |  |  |/----|`---=    |      |
      |  |                 |  |  |   ,/|==== ooo |      ;
      |  |                 |  |  |  // |(((( [33]|    ,"
      |  `-----------------'  |," .;'| |((((     |  ,"
      +-----------------------+  ;;  | |         |,"
         /_)______________(_/  //'   | +---------+
    ___________________________/___  `,
   /  oooooooooooooooo  .o.  oooo /,   `,"-----------
  / ==ooooooooooooooo==.o.  ooo= //   ,``--{)B     ,"
 /_==__==========__==_ooo__ooo=_/'   /___________,"
"""
import json
import os
import pkgutil
import re

import yaml
from jinja2 import Template
from setuptools import setup


def get_root_path(*paths):
    return os.path.join(os.path.abspath(os.path.dirname(__file__)), *paths)


def get_src_path(*paths):
    return get_root_path("src", "linktools", *paths)


class ConsoleScripts(list):

    def append_script(self, script_name, module_name):
        self.append(f"{script_name.replace('_', '-')} = {module_name}:command.main")
        return self

    def append_module(self, path, script_prefix, module_prefix):
        for _, module_name, _ in pkgutil.iter_modules([path]):
            if not module_name.startswith("_"):
                self.append_script(
                    f"{script_prefix}-{module_name.replace('_', '-')}",
                    f"{module_prefix}.{module_name}",
                )
        return self


if __name__ == '__main__':

    release = os.environ.get("RELEASE", "false").lower() == "true"
    version = os.environ.get("VERSION", None)
    if not version:
        try:
            with open(get_root_path(".version"), "rt", encoding="utf-8") as fd:
                version = fd.read().strip()
        except:
            pass
    if not version:
        version = "0.0.1"
    if version.startswith("v"):
        version = version[len("v"):]
    if not release:
        items = []
        for item in version.split("."):
            find = re.findall(r"^\d+", item)
            if find:
                items.append(int(find[0]))
        version = ".".join(map(str, items))
        version = f"{version}.post100.dev0"

    with open(get_src_path("template", "tools.yml"), "rb") as fd_in, \
            open(get_src_path("assets", "tools.json"), "wt") as fd_out:
        json.dump(
            {
                key: value
                for key, value in yaml.safe_load(fd_in).items()
                if key[0].isupper()
            },
            fd_out
        )

    with open(get_src_path("template", "metadata"), "rt", encoding="utf-8") as fd_in, \
            open(get_src_path("metadata.py"), "wt", encoding="utf-8") as fd_out:
        fd_out.write(
            Template(fd_in.read()).render(
                release=release,
                version=version,
            )
        )

    with open(get_root_path("requirements.yml"), "rt", encoding="utf-8") as fd:
        data = yaml.safe_load(fd)
        # install_requires = dependencies + dev-dependencies
        install_requires = data.get("dependencies")
        install_requires.extend(data.get("release-dependencies") if release else data.get("dev-dependencies"))
        # extras_require = optional-dependencies
        extras_require = data.get("optional-dependencies")
        all_requires = []
        for requires in extras_require.values():
            all_requires.extend(requires)
        extras_require["all"] = all_requires

    scripts = ConsoleScripts().append_script(
        script_name="lt",
        module_name="linktools.__main__",
    ).append_module(
        get_src_path("cli", "commands", "common"),
        module_prefix="linktools.cli.commands.common",
        script_prefix="ct",
    ).append_module(
        get_src_path("cli", "commands", "android"),
        module_prefix="linktools.cli.commands.android",
        script_prefix="at",
    ).append_module(
        get_src_path("cli", "commands", "ios"),
        module_prefix="linktools.cli.commands.ios",
        script_prefix="it",
    )

    setup(
        version=version,
        install_requires=install_requires,
        extras_require=extras_require,
        entry_points={"console_scripts": scripts},
    )