from setuptools import setup, find_packages

import os

import sys

#TODO: nltk, spacy data dependency

CURRENT_PYTHON = sys.version_info[:2]

REQUIRED_PYTHON = (3, 6)

# This check and everything above must remain compatible with Python 2.7.

if CURRENT_PYTHON < REQUIRED_PYTHON:

    sys.stderr.write("""

==========================

Unsupported Python version

==========================

This version of Django requires Python {}.{}, but you're trying to

install it on Python {}.{}.

This may be because you are using a version of pip that doesn't

understand the python_requires classifier. Make sure you

have pip >= 9.0 and setuptools >= 24.2, then try again:

    $ python -m pip install --upgrade pip setuptools

    $ python -m pip install django

This will install the latest version of Django which works on your

version of Python. If you can't upgrade your pip (or Python), request

an older version of Django:

    $ python -m pip install "django<2"

""".format(*(REQUIRED_PYTHON + CURRENT_PYTHON)))

    sys.exit(1)

with open('README.md') as f:

      readme = f.read()

with open('LICENSE') as f:

    license = f.read()

setup(name='ehrkit',

      version='0.1.0',

      description='NLP library for EHR data',

      url='https://github.com/Yale-LILY/EHRKit',

      author='LILY',

      install_requires=[

        'allennlp',

        'torch',

        'numpy',

        'sklearn',

        'pymysql',

        'sshtunnel',

        'nltk',

        'spacy',

        'gensim',

        'pandas',

        'bert-extractive-summarizer',

        'files2rouge',

        'scipy'

      ],

      # license='MIT',

      packages=['ehrkit', 'external', 'summarizers', 'summarizers.evaluate'],

      license=license,

      long_description=readme

)