[Install] pip install (#180)

* add bdist_wheel build

* fix wheel build

* fix setup

* fix ci

* fix setup

* fix

* fix setup

* poke ci

* fix

* poke ci

* clean build

* fix ci

* addback the env

Jenkinsfile

@@ -11,13 +11,15 @@ def setup() {
 
 def build_dgl() {
   sh "if [ -d build ]; then rm -rf build; fi; mkdir build"
-  dir("python") {
-    sh "python3 setup.py install"
-  }
   dir ("build") {
     sh "cmake .."
     sh "make -j4"
   }
+  dir("python") {
+    sh "rm -rf build *.egg-info dist"
+    sh "pip3 uninstall -y dgl"
+    sh "python3 setup.py install"
+  }
 }
 
 def pytorch_unit_test(dev) {
@@ -35,16 +37,16 @@ def mxnet_unit_test(dev) {
 }
 
 def example_test(dev) {
-  dir ("tests/scripts") {
   withEnv(["DGL_LIBRARY_PATH=${env.WORKSPACE}/build", "PYTHONPATH=${env.WORKSPACE}/python"]) {
+    dir ("tests/scripts") {
       sh "./task_example_test.sh ${dev}"
     }
   }
 }
 
 def pytorch_tutorials() {
-  dir ("tests/scripts") {
   withEnv(["DGL_LIBRARY_PATH=${env.WORKSPACE}/build", "PYTHONPATH=${env.WORKSPACE}/python"]) {
+    dir ("tests/scripts") {
       sh "./task_tutorial_test.sh"
     }
   }

python/dgl/_ffi/libinfo.py

@@ -75,7 +75,7 @@ def find_lib_path(name=None, search_path=None, optional=False):
     if not lib_found:
         message = ('Cannot find the files.\n' +
                    'List of candidates:\n' +
-                   str('\n'.join(lib_dll_path + runtime_dll_path)))
+                   str('\n'.join(lib_dll_path)))
         if not optional:
             raise RuntimeError(message)
         return None
@@ -87,4 +87,4 @@ def find_lib_path(name=None, search_path=None, optional=False):
 # We use the version of the incoming release for code
 # that is under development.
 # The following line is set by tvm/python/update_version.py
-__version__ = "0.5.dev"
+__version__ = "0.0.1"

python/setup.py

 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-import setuptools
-import os.path
+import sys,os
+import shutil
+import glob
+
+from setuptools import find_packages
+from setuptools.dist import Distribution
+from setuptools import setup
 
 with open(os.path.join(os.path.dirname(__file__), 'VERSION')) as f:
     version = f.readline().strip()
 
-setuptools.setup(
+
+class BinaryDistribution(Distribution):
+    def has_ext_modules(self):
+        return True
+
+    def is_pure(self):
+        return False
+
+CURRENT_DIR = os.path.dirname(__file__)
+
+def get_lib_path():
+    """Get library path, name and version"""
+     # We can not import `libinfo.py` in setup.py directly since __init__.py
+    # Will be invoked which introduces dependences
+    libinfo_py = os.path.join(CURRENT_DIR, './dgl/_ffi/libinfo.py')
+    libinfo = {'__file__': libinfo_py}
+    exec(compile(open(libinfo_py, "rb").read(), libinfo_py, 'exec'), libinfo, libinfo)
+    version = libinfo['__version__']
+    if not os.getenv('CONDA_BUILD'):
+        lib_path = libinfo['find_lib_path']()
+        libs = [lib_path[0]]
+    else:
+        libs = None
+    return libs, version
+
+LIBS, VERSION = get_lib_path()
+
+include_libs = False
+wheel_include_libs = False
+if not os.getenv('CONDA_BUILD'):
+    if "bdist_wheel" in sys.argv:
+        wheel_include_libs = True
+    else:
+        include_libs = True
+
+setup_kwargs = {}
+
+# For bdist_wheel only
+if wheel_include_libs:
+    with open("MANIFEST.in", "w") as fo:
+        for path in LIBS:
+            shutil.copy(path, os.path.join(CURRENT_DIR, 'dgl'))
+            _, libname = os.path.split(path)
+            fo.write("include dgl/%s\n" % libname)
+    setup_kwargs = {
+        "include_package_data": True
+    }
+
+# For source tree setup
+if include_libs:
+    rpath = [os.path.relpath(path, CURRENT_DIR) for path in LIBS]
+    setup_kwargs = {
+        "include_package_data": True,
+        "data_files": [('dgl', rpath)]
+    }
+
+setup(
     name='dgl',
     version=version,
     description='Deep Graph Library',
+    zip_safe=False,
     maintainer='DGL Team',
     maintainer_email='wmjlyjemaine@gmail.com',
-    packages=setuptools.find_packages(),
+    packages=find_packages(),
     install_requires=[
         'numpy>=1.14.0',
         'scipy>=1.1.0',
         'networkx>=2.1',
     ],
-    data_files=[('', ['VERSION'])],
-    url='https://github.com/jermainewang/dgl')
+    url='https://github.com/jermainewang/dgl',
+    distclass=BinaryDistribution,
+    classifiers=[
+        'Development Status :: 3 - Alpha',
+        'Programming Language :: Python :: 3',
+        'License :: OSI Approved :: Apache Software License',
+    ],
+    license='APACHE',
+    **setup_kwargs
+)
+
+if wheel_include_libs:
+    # Wheel cleanup
+    os.remove("MANIFEST.in")
+    for path in LIBS:
+        _, libname = os.path.split(path)
+        os.remove("dgl/%s" % libname)