dupeguru/core_pe/app.py

# Created On: 2011/09/20
# Copyright 2011 Hardcoded Software (http://www.hardcoded.net)
# 
# This software is licensed under the "BSD" License as described in the "LICENSE" file, 
# which should be included with this package. The terms are also available at 
# http://www.hardcoded.net/licenses/bsd_license

import os.path as op

from hscommon.trans import tr as trbase
from hscommon.util import format_size

from core.app import (DupeGuru as DupeGuruBase, Column, format_timestamp, format_perc,
    format_dupe_count, cmp_value)
from .scanner import ScannerPE
from . import prioritize
from . import __appname__

tr = lambda s: trbase(s, 'columns')

def format_dimensions(dimensions):
    return '%d x %d' % (dimensions[0], dimensions[1])

def get_delta_dimensions(value, ref_value):
    return (value[0]-ref_value[0], value[1]-ref_value[1])

class DupeGuru(DupeGuruBase):
    NAME = __appname__
    COLUMNS = [
        Column('name', tr("Filename")),
        Column('folder_path', tr("Folder")),
        Column('size', tr("Size (KB)")),
        Column('extension', tr("Kind")),
        Column('dimensions', tr("Dimensions")),
        Column('mtime', tr("Modification")),
        Column('percentage', tr("Match %")),
        Column('dupe_count', tr("Dupe Count")),
    ]
    DELTA_COLUMNS = {2, 4, 5}
    METADATA_TO_READ = ['size', 'mtime', 'dimensions']
    FOLDER_COL = 1
    MATCHPERC_COL = 6
    DUPECOUNT_COL = 7
    
    def __init__(self, view, appdata):
        DupeGuruBase.__init__(self, view, appdata)
        self.scanner = ScannerPE()
        self.scanner.cache_path = op.join(self.appdata, 'cached_pictures.db')
    
    def _get_display_info(self, dupe, group, delta):
        size = dupe.size
        mtime = dupe.mtime
        dimensions = dupe.dimensions
        m = group.get_match_of(dupe)
        if m:
            percentage = m.percentage
            dupe_count = 0
            if delta:
                r = group.ref
                size -= r.size
                mtime -= r.mtime
                dimensions = get_delta_dimensions(dimensions, r.dimensions)
        else:
            percentage = group.percentage
            dupe_count = len(group.dupes)
        dupe_folder_path = getattr(dupe, 'display_folder_path', dupe.folder_path)
        return [
            dupe.name,
            str(dupe_folder_path),
            format_size(size, 0, 1, False),
            dupe.extension,
            format_dimensions(dimensions),
            format_timestamp(mtime, delta and m),
            format_perc(percentage),
            format_dupe_count(dupe_count)
        ]
    
    def _get_dupe_sort_key(self, dupe, get_group, key, delta):
        if key == self.MATCHPERC_COL:
            m = get_group().get_match_of(dupe)
            return m.percentage
        if key == self.DUPECOUNT_COL:
            return 0
        if key == self.FOLDER_COL:
            dupe_folder_path = getattr(dupe, 'display_folder_path', dupe.folder_path)
            return cmp_value(str(dupe_folder_path))
        r = cmp_value(getattr(dupe, self.COLUMNS[key].attr, ''))
        if delta and (key in self.DELTA_COLUMNS):
            ref_value = cmp_value(getattr(get_group().ref, self.COLUMNS[key].attr, ''))
            if key == 4: # dimensions
                r = get_delta_dimensions(r, ref_value)
            else:
                r -= ref_value
        return r
    
    def _get_group_sort_key(self, group, key):
        if key == self.MATCHPERC_COL:
            return group.percentage
        if key == self.DUPECOUNT_COL:
            return len(group)
        if key == self.FOLDER_COL:
            dupe_folder_path = getattr(group.ref, 'display_folder_path', group.ref.folder_path)
            return cmp_value(str(dupe_folder_path))
        return cmp_value(getattr(group.ref, self.COLUMNS[key].attr, ''))
    
    def _prioritization_categories(self):
        return prioritize.all_categories()