Use walkFileTree for cleaning

Author: sebazai

tmc-langs-python3/src/main/java/fi/helsinki/cs/tmc/langs/python3/Python3Plugin.java

@@ -27,9 +27,12 @@
 
 import java.io.File;
 import java.io.IOException;
+import java.nio.file.FileVisitResult;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.nio.file.SimpleFileVisitor;
+import java.nio.file.attribute.BasicFileAttributes;
 import java.util.List;
 import java.util.Locale;
 import java.util.Map;
@@ -42,8 +45,8 @@ public final class Python3Plugin extends AbstractLanguagePlugin {
     private static final Path INIT_PY_PATH = Paths.get("__init__.py");
     private static final Path TMC_TEST_LIBRARY_PATH = Paths.get("tmc");
     private static final Path MAIN_PY_PATH = Paths.get("__main__.py");
-    private static final String CANNOT_CLEAN_FILES_MESSAGE = "Failed to clean files.";
 
+    private static final String CANNOT_CLEAN_FILES_MESSAGE = "Failed to clean files.";
     private static final String CANNOT_RUN_TESTS_MESSAGE = "Failed to run tests.";
     private static final String CANNOT_PARSE_TEST_RESULTS_MESSAGE = "Failed to read test results.";
     private static final String CANNOT_SCAN_EXERCISE_MESSAGE = "Failed to scan exercise.";
@@ -156,11 +159,29 @@ private String[] getTestCommand() {
 
     @Override
     public void clean(Path path) {
-        String[] command = {"/bin/bash", "-c", "find . -type d -name __pycache__ -exec rm -r {} \\+ -o -type f -name .available_points.json -delete -o -type f -name .tmc_test_results.json -delete" };
-        ProcessRunner runner = new ProcessRunner(command, path);
         try {
-            runner.call();
-        } catch (Exception e) {
+            Files.walkFileTree(path, new SimpleFileVisitor<Path>() {
+                @Override
+                public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) 
+                throws IOException {
+                    if (file.endsWith(".available_points.json")
+                            || file.endsWith(".tmc_test_results.json")
+                            || file.toString().contains("__pycache__")) {
+                        Files.delete(file);
+                    }
+                    return FileVisitResult.CONTINUE;
+                }
+
+                @Override
+                public FileVisitResult postVisitDirectory(Path dir, IOException exc) 
+                throws IOException {
+                    if (dir.endsWith("__pycache__")) {
+                        Files.delete(dir);
+                    }
+                    return FileVisitResult.CONTINUE;
+                }
+            });
+        } catch (IOException e) {
             log.error(CANNOT_CLEAN_FILES_MESSAGE, e);
         }
     }