dupeguru/core_me/app.py

114 lines
3.9 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
from hscommon.trans import trget
from hscommon.util import format_size, format_time
from core.app import (DupeGuru as DupeGuruBase, Column, format_timestamp,
format_perc, format_words, format_dupe_count, cmp_value)
from . import prioritize
from . import __appname__
from . import scanner, fs
coltr = trget('columns')
class DupeGuru(DupeGuruBase):
NAME = __appname__
COLUMNS = [
Column('name', coltr("Filename")),
Column('folder_path', coltr("Folder")),
Column('size', coltr("Size (MB)")),
Column('duration', coltr("Time")),
Column('bitrate', coltr("Bitrate")),
Column('samplerate', coltr("Sample Rate")),
Column('extension', coltr("Kind")),
Column('mtime', coltr("Modification")),
Column('title', coltr("Title")),
Column('artist', coltr("Artist")),
Column('album', coltr("Album")),
Column('genre', coltr("Genre")),
Column('year', coltr("Year")),
Column('track', coltr("Track Number")),
Column('comment', coltr("Comment")),
Column('percentage', coltr("Match %")),
Column('words', coltr("Words Used")),
Column('dupe_count', coltr("Dupe Count")),
]
DELTA_COLUMNS = {2, 3, 4, 5, 7}
METADATA_TO_READ = ['size', 'mtime', 'duration', 'bitrate', 'samplerate', 'title', 'artist',
'album', 'genre', 'year', 'track', 'comment']
MATCHPERC_COL = 15
DUPECOUNT_COL = 17
def __init__(self, view, appdata):
DupeGuruBase.__init__(self, view, appdata)
self.scanner = scanner.ScannerME()
self.directories.fileclasses = [fs.MusicFile]
def _get_display_info(self, dupe, group, delta):
size = dupe.size
duration = dupe.duration
bitrate = dupe.bitrate
samplerate = dupe.samplerate
mtime = dupe.mtime
m = group.get_match_of(dupe)
if m:
percentage = m.percentage
dupe_count = 0
if delta:
r = group.ref
size -= r.size
duration -= r.duration
bitrate -= r.bitrate
samplerate -= r.samplerate
mtime -= r.mtime
else:
percentage = group.percentage
dupe_count = len(group.dupes)
return [
dupe.name,
str(dupe.folder_path),
format_size(size, 2, 2, False),
format_time(duration, with_hours=False),
str(bitrate),
str(samplerate),
dupe.extension,
format_timestamp(mtime,delta and m),
dupe.title,
dupe.artist,
dupe.album,
dupe.genre,
dupe.year,
str(dupe.track),
dupe.comment,
format_perc(percentage),
format_words(dupe.words) if hasattr(dupe, 'words') else '',
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
r = cmp_value(dupe, self.COLUMNS[key])
if delta and (key in self.DELTA_COLUMNS):
r -= cmp_value(get_group().ref, self.COLUMNS[key])
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)
return cmp_value(group.ref, self.COLUMNS[key])
def _prioritization_categories(self):
return prioritize.all_categories()