setup.py

#!/usr/bin/env python

import os
import sys
import shutil
from setuptools import setup

#------------------------------------------------------------------------
# Optional building with MAVEN
#------------------------------------------------------------------------

JAVA_SRC = "knit_jvm"
jar_file = os.path.join(JAVA_SRC, "target", "knit-1.0-SNAPSHOT.jar")

with open('requirements.txt') as f:
    requires = f.read().splitlines()

if 'mvn' in sys.argv:

    os.chdir(JAVA_SRC)
    build_cmd = "mvn clean install -q"
    os.system(build_cmd)
    os.chdir("..")
    sys.argv.remove("mvn")

    java_lib_dir = os.path.join("knit", "java_libs")
    if not os.path.exists(java_lib_dir):
        os.mkdir(java_lib_dir)
    shutil.copy(jar_file, java_lib_dir)

setup(name='knit',
      version="0.2.4",
      description='Python wrapper for YARN Applications',
      url='http://github.com/dask/knit/',
      maintainer='Benjamin Zaitlen',
      maintainer_email='bzaitlen@anaconda.com',
      license='BSD',
      keywords='yarn',
      packages=['knit', 'dask_yarn'],
      package_data={'knit': ['java_libs/knit-1.0-SNAPSHOT.jar']},
      install_requires=requires,
      long_description=(open('README.rst').read()
                        if os.path.exists('README.rst') else ''),
      zip_safe=False)