Optimize remove_files()

diff --git a/mkosi/__init__.py b/mkosi/__init__.py
index 72851ecd9b8d723c18343fcc00b1501a998e1a81..45f9aef4b70a568f4726689f8d8470052a7968f5 100644 (file)
--- a/mkosi/__init__.py
+++ b/mkosi/__init__.py
@@ -130,8 +130,8 @@ def remove_files(context: Context) -> None:
         return
 
     with complete_step("Removing files…"):
-        for pattern in context.config.remove_files:
-            rmtree(*context.root.glob(pattern.lstrip("/")), sandbox=context.sandbox)
+        remove = flatten(context.root.glob(pattern.lstrip("/")) for pattern in context.config.remove_files)
+        rmtree(*remove, sandbox=context.sandbox)
 
 
 def install_distribution(context: Context) -> None: