2013-01-23 17:28:19 +00:00
|
|
|
#
|
|
|
|
# progress.py: code for handling the one big progress bar
|
|
|
|
#
|
|
|
|
# Copyright (C) 2012 Red Hat, Inc. All rights reserved.
|
|
|
|
#
|
|
|
|
# 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/>.
|
|
|
|
#
|
|
|
|
# Author(s): Chris Lumens <clumens@redhat.com>
|
|
|
|
|
|
|
|
import logging
|
|
|
|
log = logging.getLogger("anaconda")
|
|
|
|
|
|
|
|
from contextlib import contextmanager
|
2014-04-07 12:38:09 +00:00
|
|
|
|
|
|
|
from pyanaconda.queue import QueueFactory
|
2013-01-23 17:28:19 +00:00
|
|
|
|
|
|
|
# A queue to be used for communicating progress information between a subthread
|
|
|
|
# doing all the hard work and the main thread that does the GTK updates. This
|
|
|
|
# queue should have elements of the following format pushed into it:
|
|
|
|
#
|
|
|
|
# (PROGRESS_CODE_*, [arguments])
|
|
|
|
#
|
|
|
|
# Arguments vary based on the code given. See below.
|
2014-04-07 12:38:09 +00:00
|
|
|
progressQ = QueueFactory("progress")
|
2013-01-23 17:28:19 +00:00
|
|
|
|
2014-04-07 12:38:09 +00:00
|
|
|
progressQ.addMessage("init", 1) # num_steps
|
|
|
|
progressQ.addMessage("step", 0)
|
|
|
|
progressQ.addMessage("message", 1) # message
|
|
|
|
progressQ.addMessage("complete", 0)
|
|
|
|
progressQ.addMessage("quit", 1) # exit_code
|
2013-01-23 17:28:19 +00:00
|
|
|
|
|
|
|
# Surround a block of code with progress updating. Before the code runs, the
|
|
|
|
# message is updated so the user can tell what's about to take so long.
|
|
|
|
# Afterwards, the progress bar is updated to reflect that the task is done.
|
|
|
|
@contextmanager
|
|
|
|
def progress_report(message):
|
2015-03-23 11:36:12 +00:00
|
|
|
progress_message(message)
|
|
|
|
yield
|
|
|
|
progress_step(message)
|
|
|
|
|
|
|
|
def progress_message(message):
|
2014-04-07 12:38:09 +00:00
|
|
|
progressQ.send_message(message)
|
2013-01-23 17:28:19 +00:00
|
|
|
log.info(message)
|
2015-03-23 11:36:12 +00:00
|
|
|
|
|
|
|
def progress_step(message):
|
2014-04-07 12:38:09 +00:00
|
|
|
progressQ.send_step()
|
2015-03-23 11:36:12 +00:00
|
|
|
log.info(message)
|
|
|
|
|
|
|
|
def progress_init(steps):
|
|
|
|
progressQ.send_init(steps)
|
|
|
|
|
|
|
|
def progress_complete():
|
|
|
|
progressQ.send_complete()
|