summaryrefslogtreecommitdiffstats
path: root/module/gui/Queue.py
diff options
context:
space:
mode:
Diffstat (limited to 'module/gui/Queue.py')
-rw-r--r--module/gui/Queue.py252
1 files changed, 252 insertions, 0 deletions
diff --git a/module/gui/Queue.py b/module/gui/Queue.py
new file mode 100644
index 000000000..8b6f679f8
--- /dev/null
+++ b/module/gui/Queue.py
@@ -0,0 +1,252 @@
+# -*- coding: utf-8 -*-
+"""
+ 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 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 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: mkaay
+"""
+
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+
+from time import sleep, time
+
+from module.gui.Collector import CollectorModel, Package, Link, CollectorView, statusMap, statusMapReverse
+
+class QueueModel(CollectorModel):
+ def __init__(self, view, connector):
+ CollectorModel.__init__(self, view, connector)
+ self.cols = 5
+ self.wait_dict = {}
+
+ self.updater = self.QueueUpdater(self.interval)
+ self.connect(self.updater, SIGNAL("update()"), self.update)
+
+ class QueueUpdater(QObject):
+ def __init__(self, interval):
+ QObject.__init__(self)
+
+ self.interval = interval
+ self.timer = QTimer()
+ self.timer.connect(self.timer, SIGNAL("timeout()"), self, SIGNAL("update()"))
+
+ def start(self):
+ self.timer.start(1000)
+
+ def stop(self):
+ self.timer.stop()
+
+ def start(self):
+ self.updater.start()
+
+ def stop(self):
+ self.updater.stop()
+
+ def fullReload(self):
+ self._data = []
+ packs = self.connector.getPackageQueue()
+ self.beginInsertRows(QModelIndex(), 0, len(packs))
+ for pid, data in packs.items():
+ package = Package(pid, data)
+ self._data.append(package)
+ self._data = sorted(self._data, key=lambda p: p.data["order"])
+ self.endInsertRows()
+
+ def update(self):
+ locker = QMutexLocker(self.mutex)
+ downloading = self.connector.getDownloadQueue()
+ for p, pack in enumerate(self._data):
+ for d in downloading:
+ child = pack.getChild(d["id"])
+ if child:
+ child.data["downloading"] = d
+ k = pack.getChildKey(d["id"])
+ self.emit(SIGNAL("dataChanged(const QModelIndex &, const QModelIndex &)"), self.index(k, 0, self.index(p, 0)), self.index(k, self.cols, self.index(p, self.cols)))
+
+ def headerData(self, section, orientation, role=Qt.DisplayRole):
+ if orientation == Qt.Horizontal and role == Qt.DisplayRole:
+ if section == 0:
+ return QVariant(_("Name"))
+ elif section == 2:
+ return QVariant(_("Status"))
+ elif section == 1:
+ return QVariant(_("Plugin"))
+ elif section == 3:
+ return QVariant(_("Priority"))
+ elif section == 4:
+ return QVariant(_("Progress"))
+ return QVariant()
+
+ def getWaitingProgress(self, item):
+ locker = QMutexLocker(self.mutex)
+ if isinstance(item, Link):
+ if item.data["status"] == 5 and item.data["downloading"]:
+ until = float(item.data["downloading"]["wait_until"])
+ try:
+ since, until_old = self.wait_dict[item.id]
+ if not until == until_old:
+ raise Exception
+ except:
+ since = time()
+ self.wait_dict[item.id] = since, until
+ since = float(since)
+ max_wait = float(until-since)
+ rest = int(until-time())
+ res = 100/max_wait
+ perc = rest*res
+ return perc, rest
+ return None
+
+ def getProgress(self, item):
+ locker = QMutexLocker(self.mutex)
+ if isinstance(item, Link):
+ if item.data["downloading"]:
+ return int(item.data["downloading"]["percent"])
+ if item.data["statusmsg"] == "finished" or \
+ item.data["statusmsg"] == "failed" or \
+ item.data["statusmsg"] == "aborted":
+ return 100
+ elif isinstance(item, Package):
+ count = len(item.children)
+ perc_sum = 0
+ for child in item.children:
+ val = 0
+ if child.data["downloading"]:
+ val = int(child.data["downloading"]["percent"])
+ elif child.data["statusmsg"] == "finished" or \
+ child.data["statusmsg"] == "failed" or \
+ child.data["statusmsg"] == "aborted":
+ val = 100
+ perc_sum += val
+ if count == 0:
+ return 0
+ return perc_sum/count
+ return 0
+
+ def getSpeed(self, item):
+ if isinstance(item, Link):
+ if item.data["downloading"]:
+ return int(item.data["downloading"]["speed"])
+ elif isinstance(item, Package):
+ count = len(item.children)
+ speed_sum = 0
+ all_waiting = True
+ running = False
+ for child in item.children:
+ val = 0
+ if child.data["downloading"]:
+ if not child.data["statusmsg"] == "waiting":
+ all_waiting = False
+ val = int(child.data["downloading"]["speed"])
+ running = True
+ speed_sum += val
+ if count == 0 or not running or all_waiting:
+ return None
+ return speed_sum
+ return None
+
+ def data(self, index, role=Qt.DisplayRole):
+ if not index.isValid():
+ return QVariant()
+ if role == Qt.DisplayRole:
+ if index.column() == 0:
+ return QVariant(index.internalPointer().data["name"])
+ elif index.column() == 1:
+ item = index.internalPointer()
+ plugins = []
+ if isinstance(item, Package):
+ for child in item.children:
+ if not child.data["plugin"] in plugins:
+ plugins.append(child.data["plugin"])
+ else:
+ plugins.append(item.data["plugin"])
+ return QVariant(", ".join(plugins))
+ elif index.column() == 2:
+ item = index.internalPointer()
+ status = 0
+ speed = self.getSpeed(item)
+ if isinstance(item, Package):
+ for child in item.children:
+ if child.data["status"] > status:
+ status = child.data["status"]
+ else:
+ status = item.data["status"]
+
+ if speed == None or status == 7 or status == 10 or status == 5:
+ return QVariant(statusMapReverse[status])
+ else:
+ return QVariant("%s (%s KB/s)" % (statusMapReverse[status], speed))
+ elif index.column() == 3:
+ item = index.internalPointer()
+ if isinstance(item, Package):
+ return QVariant(item.data["priority"])
+ elif role == Qt.EditRole:
+ if index.column() == 0:
+ return QVariant(index.internalPointer().data["name"])
+ return QVariant()
+
+ def flags(self, index):
+ if index.column() == 0 and self.parent(index) == QModelIndex():
+ return Qt.ItemIsSelectable | Qt.ItemIsEditable | Qt.ItemIsEnabled
+ return Qt.ItemIsSelectable | Qt.ItemIsEnabled
+
+class QueueView(CollectorView):
+ def __init__(self, connector):
+ CollectorView.__init__(self, connector)
+ self.setModel(QueueModel(self, connector))
+
+ self.setColumnWidth(0, 300)
+ self.setColumnWidth(1, 100)
+ self.setColumnWidth(2, 150)
+ self.setColumnWidth(3, 50)
+ self.setColumnWidth(4, 100)
+
+ self.setEditTriggers(QAbstractItemView.NoEditTriggers)
+
+ self.delegate = QueueProgressBarDelegate(self, self.model())
+ self.setItemDelegateForColumn(4, self.delegate)
+
+class QueueProgressBarDelegate(QItemDelegate):
+ def __init__(self, parent, queue):
+ QItemDelegate.__init__(self, parent)
+ self.queue = queue
+
+ def paint(self, painter, option, index):
+ if not index.isValid():
+ return
+ if index.column() == 4:
+ item = index.internalPointer()
+ w = self.queue.getWaitingProgress(item)
+ wait = None
+ if w:
+ progress = w[0]
+ wait = w[1]
+ else:
+ progress = self.queue.getProgress(item)
+ opts = QStyleOptionProgressBarV2()
+ opts.maximum = 100
+ opts.minimum = 0
+ opts.progress = progress
+ opts.rect = option.rect
+ opts.rect.setRight(option.rect.right()-1)
+ opts.rect.setHeight(option.rect.height()-1)
+ opts.textVisible = True
+ opts.textAlignment = Qt.AlignCenter
+ if not wait == None:
+ opts.text = QString("waiting %d seconds" % (wait,))
+ else:
+ opts.text = QString.number(opts.progress) + "%"
+ QApplication.style().drawControl(QStyle.CE_ProgressBar, opts, painter)
+ return
+ QItemDelegate.paint(self, painter, option, index)
+
generated by cgit v1.2.3 (git 2.25.1) at 2025-01-24 00:07:48 +0100