8973d53cce
Add a task to calculate a trm fingerprint. * examples/trm.py (added): Add an example. Strangely enough it starts burstily, doing 10% directly, halting, then progressing quickly,
81 lines
1.9 KiB
Python
81 lines
1.9 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 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)
|
|
|
|
|
|
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)
|
|
|
|
options, args = parser.parse_args(argv[1:])
|
|
|
|
try:
|
|
path = sys.argv[1]
|
|
except IndexError:
|
|
sys.stderr.write('Please give a file to trm!\n')
|
|
return
|
|
|
|
trmtask = checksum.TRMTask(path)
|
|
|
|
if options.runner == 'cli':
|
|
runner = task.SyncRunner()
|
|
function = climain
|
|
elif options.runner == 'gtk':
|
|
runner = task.GtkProgressRunner()
|
|
function = gtkmain
|
|
|
|
function(runner, trmtask)
|
|
|
|
print
|
|
print trmtask.trm
|
|
|
|
main(sys.argv)
|