source: projects/pygtk-rpminstall/trunk/src/progress.py @ 5093

# progress-rpminstall.py
#
# Copyright (C) 2011 - Yasumichi Akahoshi
#
# This program 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 2 of the License, or
# (at your option) any later version.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.

from gi.repository import Gtk, GdkPixbuf, Gdk
import os, sys, rpm

UI_FILE = "ui/progress.ui"

class ProgressDialog:
        def __init__(self, parent, ts):
                # Build user interface
                self.builder = Gtk.Builder()
                self.builder.set_translation_domain('rpminstall')
                self.builder.add_from_file(os.path.join (os.path.dirname(__file__), UI_FILE))
                self.builder.connect_signals(self)

                self.dialog = self.builder.get_object('progress_dialog')
                self.ts = ts
                self.dialog.show()

        def run(self):
                self.ts.run (self.runCallback, None)
                return  Gtk.ResponseType.OK

        def runCallback(self, reason, amount, total, key, client_data):
                if reason == rpm.RPMCALLBACK_TRANS_START:
                        progressbar2 = self.builder.get_object('progressbar2')
                        progressbar2.set_pulse_step (1.0 / total)
                        self.total = total
                        self.step = 0
                        print "Transaction start", amount, total, key

                elif reason == rpm.RPMCALLBACK_TRANS_PROGRESS:
                        print "Transaction progress", amount, total, key

                elif reason == rpm.RPMCALLBACK_TRANS_STOP:
                        print "Transaction stop", amount, total, key

                elif reason == rpm.RPMCALLBACK_INST_OPEN_FILE:
                        print "Opening file. ", reason, amount, total, key, client_data
                        self.rpmtsCallback_fd = os.open(str(key), os.O_RDONLY)
                        label1 = self.builder.get_object('label1')
                        label1.set_text(_("Install ") + key)
                        Gtk.main_iteration()
                        return self.rpmtsCallback_fd

                elif reason == rpm.RPMCALLBACK_INST_START:
                        print "Install Start"

                elif reason == rpm.RPMCALLBACK_INST_CLOSE_FILE:
                        print "Closing file. ", reason, amount, total, key, client_data
                        os.close(self.rpmtsCallback_fd)
                        progressbar2 = self.builder.get_object('progressbar2')
                        progressbar2.pulse()
                        Gtk.main_iteration()
                        self.step += 1
                        if self.step == self.total:
                                self.dialog.destroy()

                elif reason == rpm.rpm.RPMCALLBACK_INST_PROGRESS:
                        progressbar1 = self.builder.get_object('progressbar1')
                        progressbar1.set_fraction(float(amount) / total)
                        Gtk.main_iteration()

                elif reason == rpm.RPMCALLBACK_TRANS_STOP:
                        print "stop"

                elif reason == rpm.RPMCALLBACK_UNKNOWN:
                        Gtk.main_iteration()
                        print "problem"

                elif reason == rpm.RPMCALLBACK_UNINST_PROGRESS:
                        print "RPMCALLBACK_UNINST_PROGRESS"

                elif reason == rpm.RPMCALLBACK_UNINST_START:
                        print "RPMCALLBACK_UNINST_START"

                elif reason == rpm.RPMCALLBACK_UNINST_STOP:
                        print "RPMCALLBACK_UNINST_STOP"

                elif reason == rpm.RPMCALLBACK_REPACKAGE_PROGRESS:
                        print "RPMCALLBACK_REPACKAGE_PROGRESS"

                elif reason == rpm.RPMCALLBACK_REPACKAGE_START:
                        print "RPMCALLBACK_REPACKAGE_START"

                elif reason == rpm.RPMCALLBACK_REPACKAGE_STOP:
                        print "RPMCALLBACK_REPACKAGE_STOP"

                elif reason == rpm.RPMCALLBACK_UNPACK_ERROR:
                        print "RPMCALLBACK_UNPACK_ERROR"

                elif reason == rpm.RPMCALLBACK_CPIO_ERROR:
                        print "RPMCALLBACK_CPIO_ERROR"

                else:
                        print reason