1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
|
import os
import re
from codecs import open
from setuptools import setup, find_packages
from distutils.extension import Extension
try:
from Cython.Build import cythonize
except ImportError:
USE_CYTHON = False
else:
USE_CYTHON = True
CYTHON_TRACE = bool(os.getenv('CYTHON_TRACE', False))
here = os.path.abspath(os.path.dirname(__file__))
def read_version():
regexp = re.compile(r'^VERSION\W*=\W*\(([^\(\)]*)\)')
init_py = os.path.join(here, 'clickhouse_driver', '__init__.py')
with open(init_py, encoding='utf-8') as f:
for line in f:
match = regexp.match(line)
if match is not None:
return match.group(1).replace(', ', '.')
else:
raise RuntimeError(
'Cannot find version in clickhouse_driver/__init__.py'
)
with open(os.path.join(here, 'README.rst'), encoding='utf-8') as f:
long_description = f.read()
# Prepare extensions.
ext = '.pyx' if USE_CYTHON else '.c'
extensions = [
Extension(
'clickhouse_driver.bufferedreader',
['clickhouse_driver/bufferedreader' + ext]
),
Extension(
'clickhouse_driver.bufferedwriter',
['clickhouse_driver/bufferedwriter' + ext]
),
Extension(
'clickhouse_driver.columns.largeint',
['clickhouse_driver/columns/largeint' + ext]
),
Extension(
'clickhouse_driver.varint',
['clickhouse_driver/varint' + ext]
)
]
if USE_CYTHON:
compiler_directives = {'language_level': '3'}
if CYTHON_TRACE:
compiler_directives['linetrace'] = True
extensions = cythonize(extensions, compiler_directives=compiler_directives)
github_url = 'https://github.com/mymarilyn/clickhouse-driver'
setup(
name='clickhouse-driver',
version=read_version(),
description='Python driver with native interface for ClickHouse',
long_description=long_description,
url=github_url,
author='Konstantin Lebedev',
author_email='kostyan.lebedev@gmail.com',
license='MIT',
classifiers=[
'Development Status :: 4 - Beta',
'Environment :: Console',
'Intended Audience :: Developers',
'Intended Audience :: Information Technology',
'License :: OSI Approved :: MIT License',
'Operating System :: OS Independent',
'Programming Language :: SQL',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: Implementation :: PyPy',
'Topic :: Database',
'Topic :: Software Development',
'Topic :: Software Development :: Libraries',
'Topic :: Software Development :: Libraries :: Application Frameworks',
'Topic :: Software Development :: Libraries :: Python Modules',
'Topic :: Scientific/Engineering :: Information Analysis'
],
keywords='ClickHouse db database cloud analytics',
project_urls={
'Documentation': 'https://clickhouse-driver.readthedocs.io',
'Changes': github_url + '/blob/master/CHANGELOG.md'
},
packages=find_packages('.', exclude=['tests*']),
python_requires='>=3.6, <4',
install_requires=[
'pytz',
'tzlocal'
],
ext_modules=extensions,
extras_require={
'lz4': [
'lz4<=3.0.1; implementation_name=="pypy"',
'lz4; implementation_name!="pypy"',
'clickhouse-cityhash>=1.0.2.1'
],
'zstd': ['zstd', 'clickhouse-cityhash>=1.0.2.1'],
'numpy': ['numpy>=1.12.0', 'pandas>=0.24.0']
},
test_suite='pytest'
)
|