diff --git a/PFERD/download_summary.py b/PFERD/download_summary.py index ba9d06d..b7aac4d 100644 --- a/PFERD/download_summary.py +++ b/PFERD/download_summary.py @@ -1,3 +1,8 @@ +import logging +from .logging import PrettyLogger +from .utils import PathLike + + class DownloadSummary: def __init__(self): @@ -10,19 +15,19 @@ class DownloadSummary: self._changed_files += summary._changed_files self._deleted_files += summary._deleted_files - def add_deleted_file(self, path) -> None: + def add_deleted_file(self, path: PathLike) -> None: self._deleted_files.append(path) - def add_changed_file(self, path) -> None: + def add_changed_file(self, path: PathLike) -> None: self._changed_files.append(path) - def add_new_file(self, path) -> None: + def add_new_file(self, path: PathLike) -> None: self._new_files.append(path) def _has_no_updates(self) -> bool: return len(self._new_files) == 0 and len(self._changed_files) == 0 and len(self._deleted_files) == 0 - def print(self, logger, pretty) -> None: + def print(self, logger: logging, pretty: PrettyLogger) -> None: logger.info("") logger.info("Summary: ") if self._has_no_updates(): @@ -46,5 +51,3 @@ class DownloadSummary: logger.info("Deleted Files:") for file in self._deleted_files: pretty.deleted_file(file) - - logger.info("") \ No newline at end of file