setup.py

#!/usr/bin/env python

from setuptools import setup
from setuptools.command.test import test as TestCommand
import pyhiveodbc
import sys


class PyTest(TestCommand):
    def finalize_options(self):
        TestCommand.finalize_options(self)
        self.test_args = []
        self.test_suite = True

    def run_tests(self):
        # import here, cause outside the eggs aren't loaded
        import pytest
        errno = pytest.main(self.test_args)
        sys.exit(errno)


with open('README.rst') as readme:
    long_description = readme.read()

setup(
    name="PyHiveODBC",
    version=pyhiveodbc.__version__,
    description="Python interface to Hive via ODBC",
    long_description=long_description,
    url='https://github.com/AurelienGalicher/PyHiveODBC',
    author="Aurelien Galicher",
    author_email="galicher@gmail.com",
    license="Apache License, Version 2.0",
    packages=['pyhiveodbc'],
    classifiers=[
        "Intended Audience :: Developers",
        "License :: OSI Approved :: Apache Software License",
        "Operating System :: OS Independent",
        "Topic :: Database :: Front-Ends",
    ],
    install_requires=[
        'future',
    ],
    extras_require={
        'pyodbc': ['pyodbc>=4.0.10'],
        'sqlalchemy': ['sqlalchemy>=0.8.7'],
    },
    tests_require=[
        'mock>=1.0.0',
        'pytest',
        'pytest-cov',
        'pyodbc>=4.0.10',
        'sasl>=0.2.1',
        'sqlalchemy>=0.8.7',
    ],
    cmdclass={'test': PyTest},
    package_data={
        '': ['*.rst'],
    },
    entry_points={
        'sqlalchemy.dialects': [
            'hiveodbc = pyhiveodbc.sqlalchemy_hiveodbc:HiveODBCDialect'
        ],
    }
)