fix: Handle directory/file cases when removing files

pawamoy · Oct 15, 2019 · 14114c1 · 14114c1
1 parent daea78d
commit 14114c1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/aria2p/api.py b/src/aria2p/api.py
@@ -610,7 +610,10 @@ def remove_files(downloads, force=False):
         for download in downloads:
             if download.is_complete or force:
                 for path in download.root_files_paths:
-                    shutil.rmtree(str(path))
+                    if path.is_dir():
+                        shutil.rmtree(str(path))
+                    else:
+                        path.unlink()
                 results.append(True)
             else:
                 results.append(False)