Mercurial > libervia-backend
view sat_frontends/primitivus/progress.py @ 3536:0985c47ffd96
doc: config file is now `libervia.conf` (even if `sat.conf` still works, for now)
author | Goffi <goffi@goffi.org> |
---|---|
date | Thu, 03 Jun 2021 15:15:11 +0200 |
parents | be6d91572633 |
children | 524856bd7b19 |
line wrap: on
line source
#!/usr/bin/env python3 # Primitivus: a SAT frontend # Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org) # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 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 Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. from sat.core.i18n import _ import urwid from urwid_satext import sat_widgets from sat_frontends.quick_frontend import quick_widgets class Progress(urwid.WidgetWrap, quick_widgets.QuickWidget): PROFILES_ALLOW_NONE = True def __init__(self, host, target, profiles): assert target is None and profiles is None quick_widgets.QuickWidget.__init__(self, host, target) self.host = host self.progress_list = urwid.SimpleListWalker([]) self.progress_dict = {} listbox = urwid.ListBox(self.progress_list) buttons = [] buttons.append(sat_widgets.CustomButton(_("Clear progress list"), self._onClear)) max_len = max([button.getSize() for button in buttons]) buttons_wid = urwid.GridFlow(buttons, max_len, 1, 0, "center") main_wid = sat_widgets.FocusFrame(listbox, footer=buttons_wid) urwid.WidgetWrap.__init__(self, main_wid) def add(self, progress_id, message, profile): mess_wid = urwid.Text(message) progr_wid = urwid.ProgressBar("progress_normal", "progress_complete") column = urwid.Columns([mess_wid, progr_wid]) self.progress_dict[(progress_id, profile)] = { "full": column, "progress": progr_wid, "state": "init", } self.progress_list.append(column) self.progressCB(self.host.loop, (progress_id, message, profile)) def progressCB(self, loop, data): progress_id, message, profile = data data = self.host.bridge.progressGet(progress_id, profile) pbar = self.progress_dict[(progress_id, profile)]["progress"] if data: if self.progress_dict[(progress_id, profile)]["state"] == "init": # first answer, we must construct the bar self.progress_dict[(progress_id, profile)]["state"] = "progress" pbar.done = float(data["size"]) pbar.set_completion(float(data["position"])) self.updateNotBar() else: if self.progress_dict[(progress_id, profile)]["state"] == "progress": self.progress_dict[(progress_id, profile)]["state"] = "done" pbar.set_completion(pbar.done) self.updateNotBar() return loop.set_alarm_in(0.2, self.progressCB, (progress_id, message, profile)) def _removeBar(self, progress_id, profile): wid = self.progress_dict[(progress_id, profile)]["full"] self.progress_list.remove(wid) del (self.progress_dict[(progress_id, profile)]) def _onClear(self, button): to_remove = [] for progress_id, profile in self.progress_dict: if self.progress_dict[(progress_id, profile)]["state"] == "done": to_remove.append((progress_id, profile)) for progress_id, profile in to_remove: self._removeBar(progress_id, profile) self.updateNotBar() def updateNotBar(self): if not self.progress_dict: self.host.setProgress(None) return progress = 0 nb_bars = 0 for progress_id, profile in self.progress_dict: pbar = self.progress_dict[(progress_id, profile)]["progress"] progress += pbar.current / pbar.done * 100 nb_bars += 1 av_progress = progress / float(nb_bars) self.host.setProgress(av_progress)