102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
import sys
|
|
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']
|
|
|
|
with open(os.path.join(os.path.dirname(__file__), 'rocksdb', '__init__.py'), 'r') as init_file:
|
|
version_line = [l for l in init_file.readlines() if l.startswith('__version__ = "')][0]
|
|
version = version_line.split('__version__ = "')[1][:-2]
|
|
|
|
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.12"),
|
|
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"),
|
|
]
|
|
print("✔️ all static libraries exist in expected locations")
|
|
else:
|
|
print('✘ missing static library files')
|
|
sys.exit(1)
|
|
|
|
setup(
|
|
name="lbry-rocksdb",
|
|
version=version,
|
|
keywords=['rocksdb', 'static', 'build'],
|
|
description="Python bindings for RocksDB",
|
|
long_description=open("README.md").read(),
|
|
long_description_content_type='text/markdown',
|
|
author='Jack Robison',
|
|
author_email="jackrobison@lbry.com",
|
|
url="https://github.com/lbryio/lbry-rocksdb",
|
|
license='BSD License',
|
|
python_requires=">=3.7.0",
|
|
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']
|
|
},
|
|
include_package_data=False,
|
|
zip_safe=False,
|
|
)
|