141 lines
3.5 KiB
Python
141 lines
3.5 KiB
Python
# -*- Mode: Python; test-case-name: morituri.test.test_header -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
|
|
# Morituri - for those about to RIP
|
|
|
|
# Copyright (C) 2009 Thomas Vander Stichele
|
|
|
|
# This file is part of morituri.
|
|
#
|
|
# morituri is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# morituri is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with morituri. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
import sys
|
|
import optparse
|
|
import tempfile
|
|
import pickle
|
|
import shutil
|
|
|
|
import gobject
|
|
gobject.threads_init()
|
|
import gtk
|
|
|
|
from morituri.common import checksum, task
|
|
|
|
def gtkmain(runner, taskk):
|
|
runner = task.GtkProgressRunner()
|
|
runner.connect('stop', lambda _: gtk.main_quit())
|
|
|
|
window = gtk.Window()
|
|
window.add(runner)
|
|
window.show_all()
|
|
|
|
runner.run(taskk)
|
|
|
|
gtk.main()
|
|
|
|
def climain(runner, taskk):
|
|
runner.run(taskk)
|
|
|
|
class Listener(object):
|
|
def __init__(self, path):
|
|
self.path = path
|
|
self.trms = {}
|
|
|
|
def progressed(self, task, value):
|
|
pass
|
|
|
|
def described(self, task, description):
|
|
pass
|
|
|
|
def started(self, task):
|
|
pass
|
|
|
|
def stopped(self, task):
|
|
self.trms[task.path] = task.trm
|
|
print task.path, task.trm
|
|
|
|
(fd, path) = tempfile.mkstemp(suffix='.morituri')
|
|
handle = os.fdopen(fd, 'wb')
|
|
pickle.dump(self.trms, handle, 2)
|
|
handle.close()
|
|
shutil.move(path, self.path)
|
|
|
|
|
|
def main(argv):
|
|
parser = optparse.OptionParser()
|
|
|
|
default = 'cli'
|
|
parser.add_option('-r', '--runner',
|
|
action="store", dest="runner",
|
|
help="runner ('cli' or 'gtk', defaults to %s)" % default,
|
|
default=default)
|
|
parser.add_option('-p', '--playlist',
|
|
action="store", dest="playlist",
|
|
help="playlist to analyze files from")
|
|
parser.add_option('-P', '--pickle',
|
|
action="store", dest="pickle",
|
|
help="pickle to store trms to")
|
|
|
|
|
|
options, args = parser.parse_args(argv[1:])
|
|
|
|
paths = []
|
|
if len(args) > 0:
|
|
paths.extend(args[0:])
|
|
if options.playlist:
|
|
paths.extend(open(options.playlist).readlines())
|
|
|
|
mtask = task.MultiCombinedTask()
|
|
listener = None
|
|
trms = {}
|
|
if options.pickle:
|
|
listener = Listener(options.pickle)
|
|
print 'Opening pickle %s' % options.pickle
|
|
handle = open(options.pickle)
|
|
try:
|
|
trms = pickle.load(handle)
|
|
except Exception, e:
|
|
sys.stderr.write(
|
|
"Pickle file '%s' cannot be loaded.\n" % options.pickle)
|
|
sys.exit(1)
|
|
|
|
handle.close()
|
|
|
|
for path in paths:
|
|
path = path.rstrip()
|
|
if path in trms.keys():
|
|
continue
|
|
trmtask = checksum.TRMTask(path)
|
|
if listener:
|
|
trmtask.addListener(listener)
|
|
mtask.addTask(trmtask)
|
|
mtask.description = 'Fingerprinting files'
|
|
|
|
|
|
if options.runner == 'cli':
|
|
runner = task.SyncRunner()
|
|
function = climain
|
|
elif options.runner == 'gtk':
|
|
runner = task.GtkProgressRunner()
|
|
function = gtkmain
|
|
|
|
function(runner, mtask)
|
|
|
|
print
|
|
for trmtask in mtask.tasks:
|
|
print trmtask.trm
|
|
|
|
main(sys.argv)
|