setup.py

#! /usr/bin/env python


import os
import re
import site
import sys
from distutils import sysconfig

from setuptools import find_packages, setup


def is_venv():
    real, base = map(lambda arg: hasattr(sys, arg),
                     ["real_prefix", "base_prefix"])
    return real or base and base != sys.prefix


def get_site():
    func = getattr(site, "getsitepackages", None)
    if func:
        return func()[0]
    return sysconfig.get_python_lib()


def read(fpath):
    check = os.path.isfile
    path = os.path.join(os.path.dirname(__file__), fpath)
    if not check(path):
        path = fpath
        if not check(path):
            return None
    with open(path) as stream:
        return stream.read()


def get_requirements(path="requirements.txt"):
    data = read(path)
    lines = []
    if not data:
        return lines
    for line in data.splitlines():
        line = line.strip()
        if not line or line.startswith("#"):
            continue
        if line.startswith("-r"):
            new_path = line[2:].strip()
            lines.extend(get_requirements(path=new_path))
            continue
        lines.append(line)
    return lines


def get_data_files(source, files, use_source=False):
    destination = (source if use_source else
                   os.path.join(get_site(), source))
    paths = [os.path.join(source, name) for name in files]
    return destination, paths


UNIX = bool(re.search(r"linux|darwin", sys.platform))
VENV = is_venv()
PIP = "--single-version-externally-managed" in sys.argv
LOCAL = PIP and any(map(lambda arg: arg.startswith("--home"), sys.argv))
USE_SOURCE = not UNIX or LOCAL or (VENV and not PIP)

ETC, RES = map(lambda arg: os.path.join(arg, "libmorse"), ["etc", "res"])


setup(
    name="libmorse",
    version="0.6.8",
    description="Convert timed signals into alphabet.",
    long_description=read("README.md") or "",
    url="https://github.com/cmin764/libmorse",
    license="MIT",
    author="Cosmin Poieana",
    author_email="cmin764@gmail.com",
    packages=find_packages(),
    scripts=[os.path.join("bin", "libmorse")],
    include_package_data=True,
    zip_safe=False,
    install_requires=get_requirements(),
    test_suite="tests",
    data_files = [
        get_data_files(source, files, use_source=USE_SOURCE)
        for source, files in [
            (
                ETC,
                [
                    "libmorse.conf",
                ]
            ),
            (
                RES,
                [
                    "basic_fluctuation.mor",
                    "basic.mor",
                    "basic_noise.mor",
                    "basic_slow.mor",
                    "invalid_char.mor",
                    "isolated_noise.mor",
                    "long_pause.mor",
                    "morse.json",
                    "signal_fractions.mor",
                ]
            )
        ]
    ]
)