add a script to purge installation

1 files changed, 42 insertions, 0 deletions

diff --git a/scripts/internal/purge.py b/scripts/internal/purge.py
new file mode 100755
index 00000000..d9301719
--- /dev/null
+++ b/scripts/internal/purge.py
@@ -0,0 +1,42 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2009 Giampaolo Rodola'. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""
+Purge psutil installation by removing psutil-related files and
+directories found in site-packages directories. This is needed mainly
+because sometimes "import psutil" imports a leftover installation
+from site-packages directory instead of the main working directory.
+"""
+
+import os
+import shutil
+import site
+
+
+PKGNAME = "psutil"
+
+
+def rmpath(path):
+    if os.path.isdir(path):
+        print("rmdir " + path)
+        shutil.rmtree(path)
+    else:
+        print("rm " + path)
+        os.remove(path)
+
+
+def main():
+    locations = [site.getusersitepackages()]
+    locations.extend(site.getsitepackages())
+    for root in locations:
+        if os.path.isdir(root):
+            for name in os.listdir(root):
+                if PKGNAME in name:
+                    abspath = os.path.join(root, name)
+                    rmpath(abspath)
+
+
+main()