mirror of
https://github.com/arsenetar/dupeguru.git
synced 2024-11-04 15:29:02 +00:00
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
"""
|
|
Unit Name: dupeguru.data
|
|
Created By: Virgil Dupras
|
|
Created On: 2006/03/15
|
|
Last modified by:$Author: virgil $
|
|
Last modified on:$Date: 2009-05-28 15:22:39 +0200 (Thu, 28 May 2009) $
|
|
$Revision: 4385 $
|
|
Copyright 2004-2006 Hardcoded Software (http://www.hardcoded.net)
|
|
"""
|
|
|
|
from hsutil.str import format_time, FT_DECIMAL, format_size
|
|
|
|
import time
|
|
|
|
def format_path(p):
|
|
return unicode(p[:-1])
|
|
|
|
def format_timestamp(t, delta):
|
|
if delta:
|
|
return format_time(t, FT_DECIMAL)
|
|
else:
|
|
if t > 0:
|
|
return time.strftime('%Y/%m/%d %H:%M:%S', time.localtime(t))
|
|
else:
|
|
return '---'
|
|
|
|
def format_words(w):
|
|
def do_format(w):
|
|
if isinstance(w, list):
|
|
return '(%s)' % ', '.join(do_format(item) for item in w)
|
|
else:
|
|
return w.replace('\n', ' ')
|
|
|
|
return ', '.join(do_format(item) for item in w)
|
|
|
|
def format_perc(p):
|
|
return "%0.0f" % p
|
|
|
|
def format_dupe_count(c):
|
|
return str(c) if c else '---'
|
|
|
|
def cmp_value(value):
|
|
return value.lower() if isinstance(value, basestring) else value
|
|
|
|
COLUMNS = [
|
|
{'attr':'name','display':'Filename'},
|
|
{'attr':'path','display':'Directory'},
|
|
{'attr':'size','display':'Size (KB)'},
|
|
{'attr':'extension','display':'Kind'},
|
|
{'attr':'ctime','display':'Creation'},
|
|
{'attr':'mtime','display':'Modification'},
|
|
{'attr':'percentage','display':'Match %'},
|
|
{'attr':'words','display':'Words Used'},
|
|
{'attr':'dupe_count','display':'Dupe Count'},
|
|
]
|
|
|
|
def GetDisplayInfo(dupe, group, delta=False):
|
|
if (dupe is None) or (group is None):
|
|
return ['---'] * len(COLUMNS)
|
|
size = dupe.size
|
|
ctime = dupe.ctime
|
|
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
|
|
ctime -= r.ctime
|
|
mtime -= r.mtime
|
|
else:
|
|
percentage = group.percentage
|
|
dupe_count = len(group.dupes)
|
|
return [
|
|
dupe.name,
|
|
format_path(dupe.path),
|
|
format_size(size, 0, 1, False),
|
|
dupe.extension,
|
|
format_timestamp(ctime, delta and m),
|
|
format_timestamp(mtime, delta and m),
|
|
format_perc(percentage),
|
|
format_words(dupe.words),
|
|
format_dupe_count(dupe_count)
|
|
]
|
|
|
|
def GetDupeSortKey(dupe, get_group, key, delta):
|
|
if key == 6:
|
|
m = get_group().get_match_of(dupe)
|
|
return m.percentage
|
|
if key == 8:
|
|
return 0
|
|
r = cmp_value(getattr(dupe, COLUMNS[key]['attr']))
|
|
if delta and (key in (2, 4, 5)):
|
|
r -= cmp_value(getattr(get_group().ref, COLUMNS[key]['attr']))
|
|
return r
|
|
|
|
def GetGroupSortKey(group, key):
|
|
if key == 6:
|
|
return group.percentage
|
|
if key == 8:
|
|
return len(group)
|
|
return cmp_value(getattr(group.ref, COLUMNS[key]['attr']))
|
|
|