lbry-rocksdb/setup.py

import platform
import os
from setuptools import setup
from setuptools import find_packages
from setuptools import Extension

try:
    from Cython.Build import cythonize
except ImportError:
    def cythonize(extensions):
        return extensions

    SOURCES = ['rocksdb/_rocksdb.cpp']
else:
    SOURCES = ['rocksdb/_rocksdb.pyx']

EXTRA_COMPILE_ARGS = [
    '-std=c++11',
    '-fPIC',
    '-Os',
    '-Wall',
    '-Wextra',
    '-Wconversion',
    '-fno-strict-aliasing',
    '-fno-rtti',
]

LIBRARIES = ['rocksdb', 'snappy', 'bz2', 'z', 'lz4']
EXTRA_OBJECTS = []
EXTRA_LINK_ARGS = []
INCLUDE_DIRS = []


if platform.system() == 'Darwin':
    EXTRA_COMPILE_ARGS += ['-mmacosx-version-min=10.7', '-stdlib=libc++']
    EXTRA_LINK_ARGS += ['-Wl,-s']

if platform.system() == 'Linux':
    EXTRA_LINK_ARGS += ['-Wl,--strip-all']


STATIC_LIBRARIES = [os.path.join("src", "rocksdb", item) for item in [
    "librocksdb.a",
    "libbz2.a",
    "liblz4.a",
    "libsnappy.a",
    "libz.a",
    "libzstd.a",
]]

if all(map(os.path.exists, STATIC_LIBRARIES)):
    LIBRARIES = []
    EXTRA_OBJECTS = STATIC_LIBRARIES
    INCLUDE_DIRS = [
        os.path.join("src", "rocksdb", "bzip2-1.0.8"),
        os.path.join("src", "rocksdb", "zstd-1.4.9", "lib"),
        os.path.join("src", "rocksdb", "zlib-1.2.11"),
        os.path.join("src", "rocksdb", "snappy-1.1.8"),
        os.path.join("src", "rocksdb", "snappy-1.1.8", "build"),
        os.path.join("src", "rocksdb", "lz4-1.9.3", "lib"),
        os.path.join("src", "rocksdb", "include"),
    ]


setup(
    name="python-rocksdb",
    version='0.7.0',
    keywords=['rocksdb', 'static', 'build'],
    description="Python bindings for RocksDB",
    long_description=open("README.rst").read(),
    author='Ming Hsuan Tu',
    author_email="qrnnis2623891@gmail.com",
    url="https://github.com/twmht/python-rocksdb",
    license='BSD License',
    setup_requires=['setuptools>=25', 'Cython>=0.20'],
    install_requires=['setuptools>=25'],
    package_dir={'rocksdb': 'rocksdb'},
    packages=find_packages('.'),
    ext_modules=cythonize([Extension(
        'rocksdb._rocksdb',
        SOURCES,
        extra_compile_args=EXTRA_COMPILE_ARGS,
        language='c++',
        libraries=LIBRARIES,
        include_dirs=INCLUDE_DIRS,
        extra_objects=EXTRA_OBJECTS,
        extra_link_args=EXTRA_LINK_ARGS,
    )]),
    extras_require={
        "doc": ['sphinx_rtd_theme', 'sphinx'],
        "test": ['pytest'],
    },
    include_package_data=False,
    zip_safe=False,
)