timbr-machine/setup.py at master · DigitalGlobe/timbr-machine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
from setuptools import setup
from setuptools.command.develop import develop as _develop
from setuptools.command.install import install as _install
import os

try:
    from notebook.nbextensions import install_nbextension
    from notebook.services.config import ConfigManager
except ImportError:
    install_nbextension = None
    ConfigManager = None

extension_dir = os.path.join(os.path.dirname(__file__), "timbr", "static")

class develop(_develop):
    try:
        def run(self):
            _develop.run(self)
            if install_nbextension is not None and ConfigManager is not None:
                install_nbextension(extension_dir, symlink=True,
                                overwrite=True, user=True, destination="timbr_machine")
                cm = ConfigManager()
                cm.update('notebook', {"load_extensions": {"timbr_machine/index": True } })
    except:
        pass

class install(_install):
    try:
        def run(self):
            _install.run(self)
            if install_nbextension is not None and ConfigManager is not None:
                cm = ConfigManager()
                cm.update('notebook', {"load_extensions": {"timbr_machine/index": True } })
    except:
        pass

setup(name='timbr-machine',
      cmdclass={'develop': develop, 'install': install},
      version='0.1.1',
      description='Dask-based data collection and processing machine',
      url='https://bitbucket.com/timbr-io/timbr-machine',
      author='Pramukta Kumar',
      author_email='pramukta.kumar@timbr.io',
      license='MIT',
      packages=['timbr', 'timbr.machine', 'timbr.datastore', 'timbr.compat', 'timbr.cli', 'timbr.snapshot', 'timbr.extensions'],
      zip_safe=False,
      entry_points={
        "console_scripts": [
            "machine-captd = timbr.machine.capture:main",
            "machine-snapd = timbr.machine.snapshot:main",
            "machine-pkgd = timbr.machine.packager:main",
            ]
        },
      data_files=[
        ('share/jupyter/nbextensions/timbr_machine', [
            'timbr/static/index.js'
        ])
      ],
      include_package_data=True,
      install_requires=[
          "pymongo>=2.8",
          "dask>=0.13.0",
          "ipython",
          "observed",
          "h5py",
          "jupyter_react",
          "tables>=3.2.1",
          "simplejson>=3.6.5",
          "watchdog>=0.8.1",
          "jsonpath-rw>=1.4.0",
          "pycurl",
          "rasterio>=1.0a1",
        ],
      tests_require=[
          "nose",
          "mock",
        ]
      )