Преглед на файлове

Fixes task package upgrade again

Ask Solem преди 13 години
родител
ревизия
0c040b9119
променени са 1 файла, в които са добавени 3 реда и са изтрити 2 реда
  1. 3 2
      setup.py

+ 3 - 2
setup.py

@@ -16,14 +16,15 @@ try:
         except ValueError:
             pass
     try:
-        import imp
         import celery.app
+        import imp
+        import shutil
         _, task_path, _ = imp.find_module("task", celery.app.__path__)
         if task_path.endswith("/task"):
             print("- force upgrading previous installation")
             print("  - removing %r package..." % task_path)
             try:
-                os.unlink(os.path.abspath(task_path))
+                shutil.rmtree(os.path.abspath(task_path))
             except Exception, exc:
                 sys.stderr.write("Couldn't remove %r: %r\n" % (task_path, exc))
     finally: