Add prune command to remove target and venv directories.

Author: adamreichold

x.py

@@ -2,6 +2,7 @@
 
 import argparse
 import os
+import shutil
 import subprocess
 import sys
 from pathlib import Path
@@ -24,8 +25,9 @@ def nightly():
     return b"-nightly " in proc.stdout
 
 
-def example_manifests(manifest):
-    return [dir_ / manifest for dir_ in Path("examples").iterdir()]
+def gen_examples(manifest):
+    for dir_ in Path("examples").iterdir():
+        yield dir_ / manifest
 
 
 def default(args):
@@ -36,7 +38,7 @@ def default(args):
     else:
         run("cargo", "clippy", "--all-features", "--tests")
 
-    for manifest in example_manifests("Cargo.toml"):
+    for manifest in gen_examples("Cargo.toml"):
         run("cargo", "clippy", "--manifest-path", manifest)
 
     run("cargo", "test", "--all-features", "--lib", "--tests")
@@ -55,7 +57,7 @@ def check(args):
         "warnings",
     )
 
-    for manifest in example_manifests("Cargo.toml"):
+    for manifest in gen_examples("Cargo.toml"):
         run("cargo", "fmt", "--manifest-path", manifest, "--", "--check")
 
         run("cargo", "clippy", "--manifest-path", manifest, "--", "--deny", "warnings")
@@ -97,7 +99,7 @@ def examples(args):
         sys.exit("Examples require the Nox tool (https://nox.thea.codes)")
 
     if args.name is None:
-        for manifest in example_manifests("noxfile.py"):
+        for manifest in gen_examples("noxfile.py"):
             run("nox", "--noxfile", manifest)
     else:
         run("nox", "--noxfile", f"examples/{args.name}/noxfile.py")
@@ -111,12 +113,22 @@ def format_(args):
 
     run("cargo", "fmt")
 
-    for manifest in example_manifests("Cargo.toml"):
+    for manifest in gen_examples("Cargo.toml"):
         run("cargo", "fmt", "--manifest-path", manifest)
 
     run("black", ".")
 
 
+def prune(args):
+    shutil.rmtree("target", ignore_errors=True)
+
+    for target_dir in gen_examples("target"):
+        shutil.rmtree(target_dir, ignore_errors=True)
+
+    for nox_dir in gen_examples(".nox"):
+        shutil.rmtree(nox_dir, ignore_errors=True)
+
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     subparsers = parser.add_subparsers(
@@ -159,6 +171,11 @@ def format_(args):
     )
     format_parser.set_defaults(func=format_)
 
+    prune_parser = subparsers.add_parser(
+        "prune", aliases=["p"], help="Remove target and venv directories"
+    )
+    prune_parser.set_defaults(func=prune)
+
     args = parser.parse_args()
     os.chdir(Path(__file__).parent)
     args.func(args)