forked from ibis-project/ibis
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
118 lines (103 loc) · 3.22 KB
/
setup.py
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
#!/usr/bin/env python
import pathlib
import sys
from setuptools import find_packages, setup
import versioneer
LONG_DESCRIPTION = """
Ibis is a productivity-centric Python big data framework.
See http://docs.ibis-project.org
"""
VERSION = sys.version_info.major, sys.version_info.minor
impala_requires = ['hdfs>=2.0.16', 'sqlalchemy>=1.1,<1.3.7', 'requests']
if VERSION == (3, 5):
impala_requires.append('impyla<0.14.2')
else:
impala_requires.append('impyla>=0.15.0')
sqlite_requires = ['sqlalchemy>=1.1,<1.3.7']
postgres_requires = sqlite_requires + ['psycopg2']
mysql_requires = sqlite_requires + ['pymysql']
if VERSION == (3, 5):
omniscidb_requires = ['pymapd>=0.8.3,<0.11.0']
else:
omniscidb_requires = ['pymapd>=0.12.0']
kerberos_requires = ['requests-kerberos']
visualization_requires = ['graphviz']
clickhouse_requires = ['clickhouse-driver>=0.0.8', 'clickhouse-cityhash']
bigquery_requires = ['google-cloud-bigquery>=1.0.0', 'pydata-google-auth']
hdf5_requires = ['tables>=3.0.0']
if VERSION == (3, 5):
parquet_requires = ['pyarrow<0.12.0']
else:
parquet_requires = ['pyarrow>=0.12.0']
spark_requires = ['pyspark>=2.4.3']
geospatial_requires = ['geoalchemy2', 'geopandas', 'shapely']
all_requires = (
impala_requires
+ postgres_requires
+ omniscidb_requires
+ mysql_requires
+ kerberos_requires
+ visualization_requires
+ clickhouse_requires
+ bigquery_requires
+ hdf5_requires
+ parquet_requires
+ spark_requires
+ geospatial_requires
)
develop_requires = all_requires + [
'click',
'flake8',
'isort',
'mypy',
'pre-commit',
'pygit2',
'pytest>=4.5',
]
install_requires = [
line.strip()
for line in pathlib.Path(__file__)
.parent.joinpath('requirements.txt')
.read_text()
.splitlines()
]
setup(
name='ibis-framework',
url='https://github.com/ibis-project/ibis',
packages=find_packages(),
version=versioneer.get_version(),
cmdclass=versioneer.get_cmdclass(),
install_requires=install_requires,
python_requires='>=3.5',
extras_require={
'all': all_requires,
'develop:python_version > "3.5"': develop_requires + ['black'],
'develop:python_version == "3.5"': develop_requires,
'impala': impala_requires,
'kerberos': kerberos_requires,
'postgres': postgres_requires,
'omniscidb': omniscidb_requires,
'mysql': mysql_requires,
'sqlite': sqlite_requires,
'visualization': visualization_requires,
'clickhouse': clickhouse_requires,
'bigquery': bigquery_requires,
'hdf5': hdf5_requires,
'parquet': parquet_requires,
'spark': spark_requires,
'geospatial': geospatial_requires,
},
description="Productivity-centric Python Big Data Framework",
long_description=LONG_DESCRIPTION,
classifiers=[
'Development Status :: 4 - Beta',
'Operating System :: OS Independent',
'Intended Audience :: Science/Research',
'Programming Language :: Python',
'Programming Language :: Python :: 3',
'Topic :: Scientific/Engineering',
],
license='Apache License, Version 2.0',
maintainer="Phillip Cloud",
maintainer_email="[email protected]",
)