mirror of
https://github.com/arsenetar/dupeguru.git
synced 2024-11-18 04:59:03 +00:00
95 lines
3.4 KiB
Python
95 lines
3.4 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.util import format_size
|
|
|
|
from core.app import (DupeGuru as DupeGuruBase, format_timestamp, format_perc,
|
|
format_dupe_count, cmp_value)
|
|
from .scanner import ScannerPE
|
|
from . import prioritize
|
|
from . import __appname__
|
|
from .result_table import ResultTable
|
|
|
|
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__
|
|
DELTA_COLUMNS = {2, 4, 5}
|
|
METADATA_TO_READ = ['size', 'mtime', 'dimensions']
|
|
|
|
def __init__(self, view, appdata):
|
|
DupeGuruBase.__init__(self, view, appdata)
|
|
self.scanner = ScannerPE()
|
|
self.scanner.cache_path = op.join(self.appdata, 'cached_pictures.db')
|
|
self.result_table = ResultTable(self)
|
|
|
|
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 {
|
|
'name': dupe.name,
|
|
'folder_path': str(dupe_folder_path),
|
|
'size': format_size(size, 0, 1, False),
|
|
'extension': dupe.extension,
|
|
'dimensions': format_dimensions(dimensions),
|
|
'mtime': format_timestamp(mtime, delta and m),
|
|
'percentage': format_perc(percentage),
|
|
'dupe_count': format_dupe_count(dupe_count),
|
|
}
|
|
|
|
def _get_dupe_sort_key(self, dupe, get_group, key, delta):
|
|
if key == 'percentage':
|
|
m = get_group().get_match_of(dupe)
|
|
return m.percentage
|
|
if key == 'dupe_count':
|
|
return 0
|
|
if key == 'folder_path':
|
|
dupe_folder_path = getattr(dupe, 'display_folder_path', dupe.folder_path)
|
|
return str(dupe_folder_path).lower()
|
|
r = cmp_value(dupe, key)
|
|
if delta and (key in self.DELTA_COLUMNS):
|
|
ref_value = cmp_value(get_group().ref, 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 == 'percentage':
|
|
return group.percentage
|
|
if key == 'dupe_count':
|
|
return len(group)
|
|
if key == 'folder_path':
|
|
dupe_folder_path = getattr(group.ref, 'display_folder_path', group.ref.folder_path)
|
|
return str(dupe_folder_path).lower()
|
|
return cmp_value(group.ref, key)
|
|
|
|
def _prioritization_categories(self):
|
|
return prioritize.all_categories()
|
|
|