Fix module install bug

59521dbb · wuzewu · 4784a3f2 · 59521dbb
隐藏空白更改
内联并排

Showing with 7 addition and 6 deletion

paddlehub/module/manager.py paddlehub/module/manager.py +7 -6

未找到文件。
--- a/paddlehub/module/manager.py
+++ b/paddlehub/module/manager.py
@@ -212,12 +212,13 @@ class LocalModuleManager(object):
                        fp.write(md5_value)
                save_path = os.path.join(MODULE_HOME, module_name)
-                if os.path.exists(save_path):
+                if save_path != module_dir:
-                    shutil.rmtree(save_path)
+                    if os.path.exists(save_path):
-                if from_user_dir:
+                        shutil.rmtree(save_path)
-                    shutil.copytree(module_dir, save_path)
+                    if from_user_dir:
-                else:
+                        shutil.copytree(module_dir, save_path)
-                    shutil.move(module_dir, save_path)
+                    else:
+                        shutil.move(module_dir, save_path)
                module_dir = save_path
                tips = "Successfully installed %s" % module_name
                if installed_module_version: