mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
|
"""A simple helper for managing downloaded files.
|
||
|
|
|
||
|
|
A organizer is bound to a single directory.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import List, Set
|
||
|
|
|
||
|
|
from .utils import prompt_yes_no
|
||
|
|
|
||
|
|
|
||
|
|
class Organizer():
|
||
|
|
"""A helper for managing downloaded files."""
|
||
|
|
|
||
|
|
def __init__(self, path: Path):
|
||
|
|
"""Create a new organizer for a given path."""
|
||
|
|
self._path = path
|
||
|
|
self._known_files: Set[Path] = set()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def path(self) -> Path:
|
||
|
|
"""Return the path for this organizer."""
|
||
|
|
return self._path
|
||
|
|
|
||
|
|
# TODO: Name this method :/ move_from? add_file? new_file?
|
||
|
|
def accept_file(self, source: Path, target: Path) -> None:
|
||
|
|
"""Move a file to this organizer and mark it."""
|
||
|
|
source_absolute = self.path.joinpath(source).absolute()
|
||
|
|
target_absolute = self.path.joinpath(target).absolute()
|
||
|
|
shutil.move(str(source_absolute), str(target_absolute))
|
||
|
|
self.mark_file(target)
|
||
|
|
|
||
|
|
# TODO: Name this method :/ track_file?
|
||
|
|
def mark_file(self, path: Path) -> None:
|
||
|
|
"""Mark a file as used so it will not get cleaned up."""
|
||
|
|
self._known_files.add(self.path.joinpath(path).absolute())
|
||
|
|
|
||
|
|
def resolve_file(self, file_path: Path) -> Path:
|
||
|
|
"""Resolve a file relative to the path of this organizer."""
|
||
|
|
return self.path.joinpath(file_path)
|
||
|
|
|
||
|
|
def cleanup(self) -> None:
|
||
|
|
"""Remove all untracked files in the organizer's dir."""
|
||
|
|
self._cleanup(self.path)
|
||
|
|
|
||
|
|
def _cleanup(self, start_dir: Path) -> None:
|
||
|
|
paths: List[Path] = list(start_dir.iterdir())
|
||
|
|
|
||
|
|
# Recursively clean paths
|
||
|
|
for path in paths:
|
||
|
|
if path.is_dir():
|
||
|
|
self._cleanup(path)
|
||
|
|
else:
|
||
|
|
if path.absolute() not in self._known_files:
|
||
|
|
self._delete_file_if_confirmed(path)
|
||
|
|
|
||
|
|
# Delete dir if it was empty and untracked
|
||
|
|
dir_empty = len(list(start_dir.iterdir())) == 0
|
||
|
|
if start_dir.absolute() not in self._known_files and dir_empty:
|
||
|
|
start_dir.rmdir()
|
||
|
|
|
||
|
|
def _delete_file_if_confirmed(self, path: Path) -> None:
|
||
|
|
prompt = f"Do you want to delete {path}"
|
||
|
|
|
||
|
|
if prompt_yes_no(prompt, False):
|
||
|
|
path.unlink()
|