Improve error handling for 'cleanup' command

chvancooten · Feb 28, 2023 · 63dfed9 · 63dfed9
1 parent 1aed773
commit 63dfed9
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/NimPlant.py b/NimPlant.py
@@ -251,10 +251,14 @@ def compile_nim(binary_type, xor_key, debug=False):
             if input().lower() == "y":
                 print("Cleaning up...")
 
-                os.remove("server/nimplant.db")
-                rmtree("server/downloads")
-                rmtree("server/logs")
-                rmtree("server/uploads")
+                try:
+                    os.remove("server/nimplant.db")
+                except OSError:
+                    pass
+
+                rmtree("server/downloads", ignore_errors=True)
+                rmtree("server/logs", ignore_errors=True)
+                rmtree("server/uploads", ignore_errors=True)
 
                 print("Cleaned up NimPlant server files!")
             else: