fix wrong dst path for source path under '/'

When a source file path is /FILE, its dest path would be dst/ILE.
This commit fixes this issue (#8).
This commit is contained in:
Ryo Nakamura
2024-01-11 12:53:34 +09:00
parent 4b34118a88
commit 5cbf3ad648
3 changed files with 16 additions and 6 deletions

View File

@@ -46,8 +46,10 @@ class File():
with open(self.path, "wb") as f:
f.write(os.urandom(self.size))
def cleanup(self):
def cleanup(self, preserve_dir = False):
os.remove(self.path)
if preserve_dir:
return
tmp = os.path.dirname(self.path)
while tmp and not tmp in [".", "/"]:
if len(os.listdir(tmp)) == 0: