added delete files

gramaziokohler · Nov 19, 2024 · 7414568 · 7414568
1 parent a5b9e2d
commit 7414568
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+* Added new task `DeleteFiles` to `installlib.tasks`.
+
 ### Changed
 
 ### Removed

diff --git a/src/installlib/tasks.py b/src/installlib/tasks.py
@@ -183,3 +183,20 @@ def execute(self):
         except Exception as ex:
             return False, f"Failed to copy files: {ex}"
         return True, None
+
+
+class DeleteFiles:
+    def __init__(self, filepaths, name=None) -> None:
+        self.name = name or DeleteFiles.__name__
+        self.filepaths = filepaths
+
+    def execute(self):
+        try:
+            files = self.filepaths() if callable(self.filepaths) else self.filepaths
+            if not isinstance(files, list):
+                files = [files]
+            for file in files:
+                os.remove(file)
+        except Exception as ex:
+            return False, f"Failed to delete files: {ex}"
+        return True, None