markusressel/py-image-dedup

from pathlib import Path
from typing import List

import click
from tabulate import tabulate

from py_image_dedup.library import ActionEnum
from py_image_dedup.persistence import MetadataKey
from py_image_dedup.util import echo

BYTE_IN_A_MB = 1048576


class DeduplicationResult:
    def __init__(self):
        self.item_actions = {}
        self._removed_folders = set()
        self._reference_files = {}
        self._file_duplicates = {}

    def add_file_action(self, file_path: Path, action: ActionEnum):
        if file_path in self.item_actions and self.item_actions[file_path] != action:
            raise ValueError("File path already in result "
                             "but with different action: {}, {}, {}".format(file_path,
                                                                            self.item_actions[file_path],
                                                                            action))
        self.item_actions[file_path] = action

    def get_file_with_action(self, action: ActionEnum) -> []:
        return list({k: v for k, v in self.item_actions.items() if v == action}.keys())

    def get_duplicate_count(self) -> int:
        """
        :return: amount of files that have at least one duplicate
        """
        count = 0
        for key, value in self._file_duplicates.items():
            if len(value) > 0:
                count += 1

        return count

    def get_removed_or_moved_files(self):
        return self.get_file_with_action(ActionEnum.MOVE) + self.get_file_with_action(ActionEnum.DELETE)

    def get_removed_empty_folders(self) -> []:
        """
        :return: a list of empty folders that have been deleted
        """
        return self._removed_folders

    def add_removed_empty_folder(self, folder: Path):
        """
        Adds a folder to the list of removed empty folders
        :param folder: the folder to add
        """
        self._removed_folders.add(folder)

    def set_file_duplicates(self, reference_files: List[dict], duplicate_files: []):
        """
        Set a list of files that are duplicates of the reference file
        :param reference_files: the file that is used as a baseline
        :param duplicate_files: duplicates of the reference_file
        """
        reference_file = reference_files[0]
        reference_file_path = Path(reference_file[MetadataKey.PATH.value])
        self._reference_files[reference_file_path] = reference_file
        self._file_duplicates[reference_file_path] = reference_files[1:] + duplicate_files

    def get_file_duplicates(self) -> {}:
        """
        Get a list of files that are duplicates of other files
        """
        return self._file_duplicates

    def print_to_console(self):
        title = "" * 7 + "Summary"
        echo(title, color='cyan')
        echo('=' * 21, color='cyan')
        echo(f"Files with duplicates: {self.get_duplicate_count()}")
        echo(f"Files moved: {len(self.get_file_with_action(ActionEnum.MOVE))}")
        echo(f"Files deleted: {len(self.get_file_with_action(ActionEnum.DELETE))}")

        headers = ("Action", "File path", "Dist", "Filesize", "Pixels")

        for reference_file_path, folder in self.get_file_duplicates().items():
            duplicate_count = len(folder)
            if duplicate_count > 0:
                columns = []
                echo()

                for item in [self._reference_files[reference_file_path]] + folder:
                    file_path = Path(item[MetadataKey.PATH.value])
                    distance = item[MetadataKey.DISTANCE.value]
                    distance_rounded = round(distance, 3)
                    file_size = item[MetadataKey.METADATA.value][MetadataKey.FILE_SIZE.value]
                    file_size_mb = round(file_size / BYTE_IN_A_MB, 3)
                    pixel_count = item[MetadataKey.METADATA.value][MetadataKey.PIXELCOUNT.value]

                    action = self.item_actions.get(file_path, ActionEnum.NONE)
                    row = [
                        action.name,
                        file_path,
                        distance_rounded,
                        file_size_mb,
                        pixel_count
                    ]

                    # apply action style
                    row = list(map(lambda x: str(click.style(str(x), action.color)), row))
                    columns.append(row)

                self._echo_table(
                    tabulate(columns, headers=headers, colalign=['center', 'left', 'left', 'right', 'right']))

        echo()
        echo(f"Removed (empty) folders ({len(self.get_removed_empty_folders())}):")
        for folder in self.get_removed_empty_folders():
            echo(f"{folder}", color='red')

    @staticmethod
    def _echo_table(table: str):
        lines = table.splitlines()

        for line in lines[:2]:
            echo(line, color='cyan')

        for line in lines[2:]:
            echo(line)