setup.py

# ----------------------------------------------------------------------
# Copyright (C) 2014-2015, Numenta, Inc.  Unless you have an agreement
# with Numenta, Inc., for a separate license for this software code, the
# following terms and conditions apply:
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero Public License version 3 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU Affero Public License for more details.
#
# You should have received a copy of the GNU Affero Public License
# along with this program.  If not, see http://www.gnu.org/licenses.
#
# http://numenta.org/licenses/
# ----------------------------------------------------------------------

import os
import pkg_resources
import warnings
from setuptools import setup, find_packages

REPO_DIR = os.path.dirname(os.path.realpath(__file__))



# Utility function to read the README file.
# Used for the long_description.  It's nice, because now 1) we have a top level
# README file and 2) it's easier to type in the README file than to put a raw
# string in below ...
def read(fname):
  with open(os.path.join(os.path.dirname(__file__), fname)) as f:
    result = f.read()
  return result



def nupicInstalled():
  """
  Determine whether NuPIC is already installed.
   :return: boolean
  """
  try:
    _ = pkg_resources.get_distribution("nupic")
    return True
  except pkg_resources.DistributionNotFound:
    pass  # Silently ignore. NuPIC will be installed later.

  return False



def parseFile(requirementFile):
  """
  Parse requirement file.
  :return: list of requirements.
  """
  try:
    return [
      line.strip()
      for line in open(requirementFile).readlines()
      if not line.startswith("#")
    ]
  except IOError:
    return []



def findRequirements():
  """
  Read the requirements.txt file and parse into requirements for setup's
  install_requirements option.
  """
  requirementsPath = os.path.join(REPO_DIR, "requirements.txt")
  requirements = parseFile(requirementsPath)

  if nupicInstalled():
    # The user already has a version of NuPIC installed. We'll remove the entry
    # in requirements.txt to not conflate the two and will issue a user warning.
    reqs = []
    for req in requirements:
      if "nupic" != req.split("==")[0]:
        reqs.append(req)
      else:
        warnings.warn("NuPIC is already installed so %s from requirements.txt "
                      "will be not be installed." % req)
  else:
    reqs = requirements
  return reqs



if __name__ == "__main__":
  requirements = findRequirements()

  setup(
    name="nab",
    version="1.0",
    author="Alexander Lavin",
    author_email="nab@numenta.org",
    description=(
      "Numenta Anomaly Benchmark: A benchmark for streaming anomaly prediction"),
    license="AGPL",
    packages=find_packages(),
    long_description=read("README.md"),
    install_requires=requirements,
    include_package_data=True,
    entry_points={
      "console_scripts": [
        "nab-plot = nab.plot:main",
      ],
    },
  )

  #install py2 dependencies if possible
  try:
    import os
    os.system('/bin/bash ./install_py2_detectors.sh')
  except:
    print("Unable to install python2 dependencies: numenta, numentaTM, htmjava detectors not available!")