109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
# 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 trget
|
|
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__
|
|
|
|
coltr = trget('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', coltr("Filename")),
|
|
Column('folder_path', coltr("Folder")),
|
|
Column('size', coltr("Size (KB)")),
|
|
Column('extension', coltr("Kind")),
|
|
Column('dimensions', coltr("Dimensions")),
|
|
Column('mtime', coltr("Modification")),
|
|
Column('percentage', coltr("Match %")),
|
|
Column('dupe_count', coltr("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 str(dupe_folder_path).lower()
|
|
r = cmp_value(dupe, self.COLUMNS[key])
|
|
if delta and (key in self.DELTA_COLUMNS):
|
|
ref_value = cmp_value(get_group().ref, self.COLUMNS[key])
|
|
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 str(dupe_folder_path).lower()
|
|
return cmp_value(group.ref, self.COLUMNS[key])
|
|
|
|
def _prioritization_categories(self):
|
|
return prioritize.all_categories()
|
|
|