from setuptools import setup

from downstream_node import __version__

setup(
    name='downstream-node',
    version=__version__,
    packages=['downstream_node'],
    url='',
    license='',
    author='Storj Labs',
    author_email='info@storj.io',
    description='',
    install_requires=[
        'flask',
        'mysql-python',
        'flask-sqlalchemy',
        'heartbeat==0.1.2',
    ],
    dependency_links=[
        'https://github.com/Storj/heartbeat/archive/v0.1.2.tar.gz#egg=heartbeat-0.1.2'
    ],
)

import sys
from setuptools import setup

from downstream_node import __version__

# Reqirements for all versions of Python
install_requires = [
    'flask',
    'pymysql',
    'flask-sqlalchemy',
    'heartbeat==0.1.2',
]

# Requirements for Python 2
if sys.version_info < (3,):
    extras = [
        'mysql-python',
    ]
    install_requires.extend(extras)

setup(
    name='downstream-node',
    version=__version__,
    packages=['downstream_node'],
    url='',
    license='',
    author='Storj Labs',
    author_email='info@storj.io',
    description='',
    install_requires=install_requires,
    dependency_links=[
        'git+https://github.com/Storj/heartbeat.git@v0.1.2#egg=heartbeat-0.1.2'
    ],
)

Add proper dependency links and install_requires lines for Py3 and Py2 support