From 9d54734c7ea4fabdb445c60b9b7738f1edade330 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Wed, 10 Jun 2020 15:47:38 +0800
Subject: [PATCH] windows supported

---
 setup.py | 41 +++++++++++++++++++++--------------------
 1 file changed, 21 insertions(+), 20 deletions(-)

diff --git a/setup.py b/setup.py
index b3e3a362..5720cccc 100644
--- a/setup.py
+++ b/setup.py
@@ -34,27 +34,28 @@ about["__url__"] = "https://github.com/PaddlePaddle/PaddleRec"
 readme = ""
 
 
-def run_cmd(command):
-    assert command is not None and isinstance(command, str)
-    return os.popen(command).read().strip()
-
-
 def build(dirname):
     package_dir = os.path.dirname(os.path.abspath(__file__))
-    run_cmd("cp -r {}/* {}".format(package_dir, dirname))
-    run_cmd("mkdir {}".format(os.path.join(dirname, "paddlerec")))
-    run_cmd("mv {} {}".format(
-        os.path.join(dirname, "core"), os.path.join(dirname, "paddlerec")))
-    run_cmd("mv {} {}".format(
-        os.path.join(dirname, "doc"), os.path.join(dirname, "paddlerec")))
-    run_cmd("mv {} {}".format(
-        os.path.join(dirname, "models"), os.path.join(dirname, "paddlerec")))
-    run_cmd("mv {} {}".format(
-        os.path.join(dirname, "tests"), os.path.join(dirname, "paddlerec")))
-    run_cmd("mv {} {}".format(
-        os.path.join(dirname, "tools"), os.path.join(dirname, "paddlerec")))
-    run_cmd("mv {} {}".format(
-        os.path.join(dirname, "*.py"), os.path.join(dirname, "paddlerec")))
+    shutil.copytree(
+        package_dir, dirname, ignore=shutil.ignore_patterns(".git"))
+    os.mkdir(os.path.join(dirname, "paddlerec"))
+    shutil.move(
+        os.path.join(dirname, "core"), os.path.join(dirname, "paddlerec"))
+    shutil.move(
+        os.path.join(dirname, "doc"), os.path.join(dirname, "paddlerec"))
+    shutil.move(
+        os.path.join(dirname, "models"), os.path.join(dirname, "paddlerec"))
+    shutil.move(
+        os.path.join(dirname, "tests"), os.path.join(dirname, "paddlerec"))
+    shutil.move(
+        os.path.join(dirname, "tools"), os.path.join(dirname, "paddlerec"))
+
+    for f in os.listdir(dirname):
+        if os.path.isdir(f):
+            continue
+        if os.path.splitext(f)[1] == ".py":
+            shutil.move(
+                os.path.join(dirname, f), os.path.join(dirname, "paddlerec"))
 
     packages = find_packages(dirname, include=('paddlerec.*'))
     package_dir = {'': dirname}
@@ -90,7 +91,7 @@ def build(dirname):
         zip_safe=False)
 
 
-dirname = tempfile.mkdtemp()
+dirname = tempfile.mktemp()
 build(dirname)
 shutil.rmtree(dirname)
 
-- 
GitLab