From 14114c1fdb4a7de6ca4d24e488151a7b6864bb94 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Timoth=C3=A9e=20Mazzucotelli?= Date: Tue, 15 Oct 2019 22:12:49 +0200 Subject: [PATCH] fix: Handle directory/file cases when removing files --- src/aria2p/api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/aria2p/api.py b/src/aria2p/api.py index bfaa6a0..be3b075 100644 --- 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)