From 7cf8e0c9e6a619cc27c678cd2624c74717ad51ce Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 6 Sep 2016 20:09:52 +0800
Subject: [PATCH] Lazy install Paddle wheels

* install wheels when invoke paddle script if current python don't
  have paddle packages, or installed a previous version.
* Also add `make install` to travis
---
 .travis.yml                           |  1 +
 paddle/scripts/submit_local.sh.in     | 34 +++++++++++++++++++++++++++
 paddle/scripts/travis/build.sh        |  6 +----
 paddle/scripts/travis/common.sh       |  5 ++++
 paddle/scripts/travis/make_install.sh |  5 ++++
 paddle/scripts/travis/unittest.sh     |  3 +--
 python/CMakeLists.txt                 |  4 ----
 7 files changed, 47 insertions(+), 11 deletions(-)
 create mode 100755 paddle/scripts/travis/common.sh
 create mode 100755 paddle/scripts/travis/make_install.sh

diff --git a/.travis.yml b/.travis.yml
index 644b8dfb236..a78853e15b1 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -29,6 +29,7 @@ before_install:
 script:
   - paddle/scripts/travis/build.sh
   - paddle/scripts/travis/unittest.sh
+  - paddle/scripts/travis/make_install.sh
 notifications:
   email:
     on_success: change
diff --git a/paddle/scripts/submit_local.sh.in b/paddle/scripts/submit_local.sh.in
index eed2d315932..6d2cab6b16b 100644
--- a/paddle/scripts/submit_local.sh.in
+++ b/paddle/scripts/submit_local.sh.in
@@ -43,6 +43,40 @@ fi
 
 export PYTHONPATH=${PWD}:${PYTHONPATH}
 
+
+# Check python lib installed or not.
+pip --help > /dev/null
+if [ $? -ne 0 ]; then
+    echo "pip should be installed to run paddle."
+    exit 1
+fi
+
+INSTALLED_VERSION=`pip freeze 2>/dev/null | grep '^paddle' | sed 's/.*==//g'`
+
+if [ -z ${INSTALLED_VERSION} ]; then
+   INSTALLED_VERSION="0.0.0"  # not installed
+fi
+cat <<EOF | python -
+from distutils.version import LooseVersion
+import sys
+if LooseVersion("${INSTALLED_VERSION}") < LooseVersion("@PADDLE_VERSION@"):
+  sys.exit(1)
+else:
+  sys.exit(0)
+EOF
+
+if [ $? -eq 1 ]; then  # Older version installed, or not installed at all
+   echo "First time run paddle, need to install some python dependencies."
+   BASEDIR=$(dirname "$0")
+   pip install ${BASEDIR}/../opt/paddle/share/wheels/*.whl
+   if [ $? -ne 0 ]; then
+      echo "pip install wheels failed. "
+      echo "Please use 'sudo paddle' at the first time you use PaddlePaddle"
+      echo "PaddlePaddle will install some python dependencies automatically."
+   fi
+   echo "Python dependencies are installed."
+fi
+
 case "$1" in
     "train")
         ${DEBUGGER} $MYDIR/../opt/paddle/bin/paddle_trainer ${@:2}
diff --git a/paddle/scripts/travis/build.sh b/paddle/scripts/travis/build.sh
index 638b1bafb43..a644f2a4164 100755
--- a/paddle/scripts/travis/build.sh
+++ b/paddle/scripts/travis/build.sh
@@ -1,9 +1,5 @@
 #!/bin/bash
-
 cd `dirname $0`
-cd ../../../
-set -e
-mkdir build
-cd build
+source ./common.sh
 cmake .. -DCMAKE_BUILD_TYPE=Debug -DWITH_GPU=OFF -DWITH_DOC=OFF -DWITH_TESTING=ON -DON_TRAVIS=ON
 make -j `nproc`
diff --git a/paddle/scripts/travis/common.sh b/paddle/scripts/travis/common.sh
new file mode 100755
index 00000000000..37e27d665b1
--- /dev/null
+++ b/paddle/scripts/travis/common.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+set -e
+mkdir -p ../../../build
+cd ../../../build
+
diff --git a/paddle/scripts/travis/make_install.sh b/paddle/scripts/travis/make_install.sh
new file mode 100755
index 00000000000..08b2a648bb9
--- /dev/null
+++ b/paddle/scripts/travis/make_install.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+cd `dirname $0`
+source ./common.sh
+sudo make install
+sudo paddle version
diff --git a/paddle/scripts/travis/unittest.sh b/paddle/scripts/travis/unittest.sh
index a095e308f32..45e8c85c102 100755
--- a/paddle/scripts/travis/unittest.sh
+++ b/paddle/scripts/travis/unittest.sh
@@ -1,6 +1,5 @@
 #!/bin/bash
-set -e
 cd `dirname $0`
-cd ../../../build
+source ./common.sh
 env CTEST_OUTPUT_ON_FAILURE=1 make test ARGS="-j `nproc`"
 
diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
index c8e3ecd41ce..68cc4024704 100644
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@@ -22,10 +22,6 @@ find_python_module(pip REQUIRED)
 find_python_module(wheel REQUIRED)
 find_python_module(google.protobuf REQUIRED)
 
-install(CODE "execute_process(COMMAND ${PYTHON_EXECUTABLE} setup.py install -f
-    WORKING_DIRECTORY
-    ${CMAKE_CURRENT_BINARY_DIR})")
-
 install(DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/dist/
     DESTINATION opt/paddle/share/wheels
 )
-- 
GitLab