forked from ibis-project/ibis
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
106 lines (92 loc) · 2.82 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
#!/usr/bin/env python
"""Ibis setup module."""
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://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']
impala_requires.append('impyla[kerberos]>=0.15.0')
sqlite_requires = ['sqlalchemy>=1.1,<1.3.7']
postgres_requires = sqlite_requires + ['psycopg2']
mysql_requires = sqlite_requires + ['pymysql']
kerberos_requires = ['requests-kerberos']
visualization_requires = ['graphviz']
clickhouse_requires = [
'clickhouse-driver>=0.1.3',
'clickhouse-cityhash',
]
bigquery_requires = [
'google-cloud-bigquery[bqstorage,pandas]>=1.12.0,<2.0.0dev',
'pydata-google-auth',
]
hdf5_requires = ['tables>=3.0.0']
parquet_requires = ['pyarrow>=0.12.0']
spark_requires = ['pyspark>=2.4.3']
geospatial_requires = ['geoalchemy2', 'geopandas', 'shapely']
dask_requires = [
'dask[dataframe, array]>=2.22.0',
]
all_requires = (
impala_requires
+ postgres_requires
+ mysql_requires
+ kerberos_requires
+ visualization_requires
+ clickhouse_requires
+ bigquery_requires
+ hdf5_requires
+ parquet_requires
+ spark_requires
+ geospatial_requires
+ dask_requires
)
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.7',
extras_require={
'all': all_requires,
'impala': impala_requires,
'kerberos': kerberos_requires,
'postgres': postgres_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,
'dask': dask_requires,
},
description="Productivity-centric Python Big Data Framework",
long_description=LONG_DESCRIPTION,
classifiers=[
'Development Status :: 5 - Production/Stable',
'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]",
)