477 Move clear_comtypes_cache to be a callable module

comtypes/clear_cache.py

@@ -0,0 +1,43 @@
+import argparse
+import os
+import sys
+from shutil import rmtree  # TESTS ASSUME USE OF RMTREE
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="py -m comtypes.clear_cache", description="Removes comtypes cache folders."
+    )
+    parser.add_argument(
+        "-y", help="Pre-approve deleting all folders", action="store_true"
+    )
+    args = parser.parse_args()
+
+    if not args.y:
+        confirm = input("Remove comtypes cache directories? (y/n): ")
+        if confirm.lower() != "y":
+            print("Cache directories NOT removed")
+            return
+
+    # change cwd to avoid import from local folder during installation process
+    work_dir = os.getcwd()
+    try:
+        os.chdir(os.path.dirname(sys.executable))
+        import comtypes.client
+    except ImportError:
+        print("Could not import comtypes", file=sys.stderr)
+        sys.exit(1)
+    os.chdir(work_dir)
+
+    # there are two possible locations for the cache folder (in the comtypes
+    # folder in site-packages if that is writable, otherwise in APPDATA)
+    # fortunately, by deleting the first location returned by _find_gen_dir()
+    # we make it un-writable, so calling it again gives us the APPDATA location
+    for _ in range(2):
+        dir_path = comtypes.client._find_gen_dir()
+        rmtree(dir_path)
+        print(f'Removed directory "{dir_path}"')
+
+
+if __name__ == "__main__":
+    main()

comtypes/test/test_clear_cache.py

@@ -0,0 +1,22 @@
+"""
+Test for the ``comtypes.clear_cache`` module.
+"""
+import io
+import runpy
+from unittest.mock import patch, call
+from unittest import TestCase
+
+from comtypes.client import _find_gen_dir
+
+
+class ClearCacheTestCase(TestCase):
+    # we patch sys.stdout so unittest doesn't show the print statements
+    @patch("sys.stdout", new=io.StringIO())
+    @patch("shutil.rmtree")
+    def test_clear_cache(self, mock_rmtree):
+        with patch("sys.argv", ["clear_cache.py", "-y"]):
+            runpy.run_module("comtypes.clear_cache", {}, "__main__")
+
+        # because we don't actually delete anything, _find_gen_dir() will
+        # give the same answer every time we call it
+        assert mock_rmtree.call_args_list == [call(_find_gen_dir()) for i in range(2)]

setup.py

@@ -111,14 +111,10 @@ def run(self):
         install.run(self)
         # Custom script we run at the end of installing
         if not self.dry_run and not self.root:
-            filename = os.path.join(
-                self.install_scripts, "clear_comtypes_cache.py")
-            if not os.path.isfile(filename):
-                raise RuntimeError("Can't find '%s'" % (filename,))
             print("Executing post install script...")
-            print('"' + sys.executable + '" "' + filename + '" -y')
+            print('"' + sys.executable + '" -m comtypes.clear_cache -y')
             try:
-                subprocess.check_call([sys.executable, filename, '-y'])
+                subprocess.check_call([sys.executable, "-m", "comtypes.clear_cache", '-y'])
             except subprocess.CalledProcessError:
                 print("Failed to run post install script!")
 
@@ -146,7 +142,9 @@ def run(self):
         ]},
     classifiers=classifiers,
 
-    scripts=["clear_comtypes_cache.py"],
+    entry_points={
+        "console_scripts": ["clear_comtypes_cache=comtypes.clear_cache:main"]
+    },
 
     cmdclass={
         'test': test,