proxy.py/proxy/common/_version.py

# -*- coding: utf-8 -*-
"""
    proxy.py
    ~~~~~~~~
    ⚡⚡⚡ Fast, Lightweight, Pluggable, TLS interception capable proxy server focused on
    Network monitoring, controls & Application development, testing, debugging.

    :copyright: (c) 2013-present by Abhinav Singh and contributors.
    :license: BSD, see LICENSE for more details.

    Version definition.
"""
from typing import Tuple, Union


def _get_dist(distribution_name: str) -> str:
    # pylint: disable=import-outside-toplevel
    import warnings

    try:
        # pylint: disable=import-outside-toplevel
        from importlib.metadata import version  # noqa: WPS433

        return version(distribution_name)
    except ModuleNotFoundError:  # pragma: no cover
        with warnings.catch_warnings():
            warnings.filterwarnings('ignore', category=DeprecationWarning)

            # pylint: disable=import-outside-toplevel
            from pkg_resources import get_distribution  # noqa: WPS433

            return get_distribution(distribution_name).version


try:
    # pylint: disable=unused-import
    from ._scm_version import version as __version__  # noqa: WPS433, WPS436
    from ._scm_version import version_tuple as _ver_tup  # noqa: WPS433, WPS436
except ImportError:  # pragma: no cover
    __version__ = _get_dist('proxy.py')  # noqa: WPS440


def _to_int_or_str(inp: str) -> Union[int, str]:    # pragma: no cover
    try:
        return int(inp)
    except ValueError:
        return inp


def _split_version_parts(inp: str) -> Tuple[str, ...]:  # pragma: no cover
    public_version, _plus, local_version = inp.partition('+')
    return (*public_version.split('.'), local_version)


try:
    VERSION = _ver_tup
except NameError:   # pragma: no cover
    VERSION = tuple(
        map(_to_int_or_str, _split_version_parts(__version__)),
    )


__all__ = '__version__', 'VERSION'