diff --git a/melamine/shred.py b/melamine/shred.py index 375c0fc..67f0bd6 100644 --- a/melamine/shred.py +++ b/melamine/shred.py @@ -1,6 +1,6 @@ from .classes import ShredDir from .classes import ShredFile -from .fileops import mount_to_fs +from .fileops import mount_to_fs_handler from .logs import logger @@ -16,13 +16,13 @@ async def main(job) -> bool: if path.is_file(): logger.info(f"Adding file: {path}") shred_file = ShredFile(path) - shred_file.fs_handler = await mount_to_fs(shred_file.mount_point) + shred_file.fs_handler = await mount_to_fs_handler(shred_file.mount_point) new_paths.add(shred_file) elif path.is_dir(): if job.recursive: logger.info(f"Adding directory: {path}") shred_dir = ShredDir(path) - shred_dir.fs_handler = await mount_to_fs(shred_dir.mount_point) + shred_dir.fs_handler = await mount_to_fs_handler(shred_dir.mount_point) new_paths.add(shred_dir) else: logger.info(f"Skipping directory: {path} (try -r/--recursive)")