LBRYCommunity/lbry-rocksdb
1
0
Fork 0
Code Issues 6 Pull requests 2 Projects Releases Packages Wiki Activity

Update setup.py

Browse source
This commit is contained in:
iFA 2021-08-26 10:37:33 +02:00 • committed by GitHub
parent a93d782ad7
commit 6e8c23f794
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 60 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

74
setup.py
View file

@ -1,12 +1,23 @@
import platform, sys import platform
import os
from setuptools import setup from setuptools import setup
from setuptools import find_packages from setuptools import find_packages
from setuptools import Extension from setuptools import Extension
try:
from Cython.Build import cythonize
except ImportError:
def cythonize(extensions):
return extensions
extra_compile_args = [ SOURCES = ['rocksdb/_rocksdb.cpp']
else:
SOURCES = ['rocksdb/_rocksdb.pyx']
EXTRA_COMPILE_ARGS = [
'-std=c++11', '-std=c++11',
'-O3', '-fPIC',
'-Os',
'-Wall', '-Wall',
'-Wextra', '-Wextra',
'-Wconversion', '-Wconversion',
@ -14,17 +25,49 @@ extra_compile_args = [
'-fno-rtti', '-fno-rtti',
] ]
if platform.system() == 'Darwin': LIBRARIES = ['rocksdb', 'snappy', 'bz2', 'z', 'lz4']
extra_compile_args += ['-mmacosx-version-min=10.7', '-stdlib=libc++'] 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"),
]
if sys.version_info < (3 , 0):
raise Exception("pyRocksDB require Python 3.x")
setup( setup(
name="python-rocksdb", name="python-rocksdb",
version='0.7.0', version='0.7.0',
keywords=['rocksdb', 'static', 'build'],
description="Python bindings for RocksDB", description="Python bindings for RocksDB",
keywords='rocksdb', long_description=open("README.rst").read(),
author='Ming Hsuan Tu', author='Ming Hsuan Tu',
author_email="qrnnis2623891@gmail.com", author_email="qrnnis2623891@gmail.com",
url="https://github.com/twmht/python-rocksdb", url="https://github.com/twmht/python-rocksdb",
@ -33,17 +76,20 @@ setup(
install_requires=['setuptools>=25'], install_requires=['setuptools>=25'],
package_dir={'rocksdb': 'rocksdb'}, package_dir={'rocksdb': 'rocksdb'},
packages=find_packages('.'), packages=find_packages('.'),
ext_modules=[Extension( ext_modules=cythonize([Extension(
'rocksdb._rocksdb', 'rocksdb._rocksdb',
['rocksdb/_rocksdb.pyx'], SOURCES,
extra_compile_args=extra_compile_args, extra_compile_args=EXTRA_COMPILE_ARGS,
language='c++', language='c++',
libraries=['rocksdb', 'snappy', 'bz2', 'z', 'lz4'], libraries=LIBRARIES,
)], include_dirs=INCLUDE_DIRS,
extra_objects=EXTRA_OBJECTS,
extra_link_args=EXTRA_LINK_ARGS,
)]),
extras_require={ extras_require={
"doc": ['sphinx_rtd_theme', 'sphinx'], "doc": ['sphinx_rtd_theme', 'sphinx'],
"test": ['pytest'], "test": ['pytest'],
}, },
include_package_data=True, include_package_data=False,
zip_safe=False, zip_safe=False,
) )