source: orange/Orange/OrangeWidgets/OWDatabasesUpdate.py @ 11739:53aea1191799

Revision 11739:53aea1191799, 27.4 KB checked in by Ales Erjavec <ales.erjavec@…>, 6 months ago (diff)

Fixed default view column sizes.

Line 
1from __future__ import with_statement
2
3import os
4import sys
5
6from datetime import datetime, timedelta
7from functools import partial
8from collections import namedtuple
9
10from PyQt4.QtCore import pyqtSignal as Signal, pyqtSlot as Slot
11
12from Orange.utils import serverfiles, environ
13from Orange.utils.serverfiles import sizeformat as sizeof_fmt
14
15from OWWidget import *
16
17from OWConcurrent import Task, ThreadExecutor, methodinvoke
18
19import OWGUIEx
20import OWGUI
21
22
23#: Update file item states
24AVAILABLE, CURRENT, OUTDATED, DEPRECATED = range(4)
25
26_icons_dir = os.path.join(environ.canvas_install_dir, "icons")
27
28
29def icon(name):
30    return QIcon(os.path.join(_icons_dir, name))
31
32
33class ItemProgressBar(QProgressBar):
34    """Progress Bar with and `advance()` slot.
35    """
36    @Slot()
37    def advance(self):
38        """
39        Advance the progress bar by 1
40        """
41        self.setValue(self.value() + 1)
42
43
44class UpdateOptionButton(QToolButton):
45    def event(self, event):
46        if event.type() == QEvent.Wheel:
47            # QAbstractButton automatically accepts all mouse events (in
48            # event method) for disabled buttons. This can prevent scrolling
49            # in a scroll area when a disabled button scrolls under the
50            # mouse.
51            event.ignore()
52            return False
53        else:
54            return QToolButton.event(self, event)
55
56
57class UpdateOptionsWidget(QWidget):
58    """
59    A Widget with download/update/remove options.
60    """
61    #: Install/update button was clicked
62    installClicked = Signal()
63    #: Remove button was clicked.
64    removeClicked = Signal()
65   
66    def __init__(self, state=AVAILABLE, parent=None):
67        QWidget.__init__(self, parent)
68        layout = QHBoxLayout()
69        layout.setSpacing(1)
70        layout.setContentsMargins(1, 1, 1, 1)
71
72        self.checkButton = QCheckBox()
73         
74        layout.addWidget(self.checkButton)
75        self.setLayout(layout)
76 
77        self.setMaximumHeight(30)
78
79        self.state = -1
80        self.setState(state)
81       
82    def setState(self, state):
83        """
84        Set the current update state for the widget (AVAILABLE,
85        CURRENT, OUTDATED or DEPRECTED).
86
87        """
88        if self.state != state:
89            self.state = state
90            self._update()
91 
92    def _update(self):
93        if self.state == AVAILABLE:
94            self.checkButton.setChecked(False)
95        elif self.state == CURRENT:
96            self.checkButton.setChecked(True)
97        elif self.state == OUTDATED:
98            self.checkButton.setChecked(True)
99        elif self.state == DEPRECATED:
100            self.checkButton.setChecked(True)
101        else:
102            raise ValueError("Invalid state %r" % self._state)
103
104        try:
105            self.checkButton.clicked.disconnect()   # Remove old signals if they exist
106        except:
107            pass
108
109        if not self.checkButton.isChecked():        # Switch signals if the file is present or not
110            self.checkButton.clicked.connect(self.installClicked)
111        else:
112            self.checkButton.clicked.connect(self.removeClicked)
113
114
115class UpdateTreeWidgetItem(QTreeWidgetItem):
116    """
117    A QTreeWidgetItem for displaying an UpdateItem.
118
119    :param UpdateItem item:
120        The update item for display.
121
122    """
123    STATE_STRINGS = {0: "not downloaded",
124                     1: "downloaded, current",
125                     2: "downloaded, needs update",
126                     3: "obsolete"}
127
128    #: A role for the state item data.
129    StateRole = OWGUI.OrangeUserRole.next()
130
131    # QTreeWidgetItem stores the DisplayRole and EditRole as the same role,
132    # so we can't use EditRole to store the actual item data, instead we use
133    # custom role.
134
135    #: A custom edit role for the item's data
136    EditRole2 = OWGUI.OrangeUserRole.next()
137
138    def __init__(self, item):
139        QTreeWidgetItem.__init__(self, type=QTreeWidgetItem.UserType)
140
141        self.item = None
142        self.setUpdateItem(item)
143
144    def setUpdateItem(self, item):
145        """
146        Set the update item for display.
147
148        :param UpdateItem item:
149            The update item for display.
150
151        """
152        self.item = item
153
154        self.setData(0, UpdateTreeWidgetItem.StateRole, item.state)
155
156        self.setData(1, Qt.DisplayRole, item.title)
157        self.setData(1, self.EditRole2, item.title)
158
159        self.setData(4, Qt.DisplayRole, sizeof_fmt(item.size))
160        self.setData(4, self.EditRole2, item.size)
161
162        if item.local is not None:
163            self.setData(3, Qt.DisplayRole, item.local.date().isoformat())
164            self.setData(3, self.EditRole2, item.local)
165        else:
166            self.setData(3, Qt.DisplayRole, "")
167            self.setData(3, self.EditRole2, datetime.now())
168
169        self._updateToolTip()
170
171    def _updateToolTip(self):
172        state_str = self.STATE_STRINGS[self.item.state]
173        try:
174            diff_date = self.item.latest - self.item.local
175        except:
176            diff_date = None
177       
178        tooltip = ("State: %s\nTags: %s" %
179                   (state_str, ", ".join(tag for tag in self.item.tags
180                    if not tag.startswith("#"))))
181
182        if self.item.state in [CURRENT, OUTDATED, DEPRECATED]:
183            tooltip += ("\nFile: %s" %
184                        serverfiles.localpath(self.item.domain,
185                                              self.item.filename))
186       
187        if self.item.state == OUTDATED and diff_date:
188            tooltip += ("\nServer version: %s\nStatus: old (%d days)" % (self.item.latest, diff_date.days))
189        else:
190            tooltip += ("\nServer version: %s" % self.item.latest)
191
192        for i in range(1, 4):
193            self.setToolTip(i, tooltip)
194
195    def __lt__(self, other):
196        widget = self.treeWidget()
197        column = widget.sortColumn()
198        if column == 0:
199            role = UpdateTreeWidgetItem.StateRole
200        else:
201            role = self.EditRole2
202
203        left = self.data(column, role).toPyObject()
204        right = other.data(column, role).toPyObject()
205        return left < right
206
207
208class UpdateOptionsItemDelegate(QStyledItemDelegate):
209    """
210    An item delegate for the updates tree widget.
211
212    .. note: Must be a child of a QTreeWidget.
213
214    """
215    def sizeHint(self, option, index):
216        size = QStyledItemDelegate.sizeHint(self,  option, index)
217        parent = self.parent()
218        item = parent.itemFromIndex(index)
219        widget = parent.itemWidget(item, 0)
220        if widget:
221            size = QSize(size.width(), widget.sizeHint().height() / 2)
222        return size
223
224
225UpdateItem = namedtuple(
226    "UpdateItem",
227    ["domain",
228     "filename",
229     "state",  # Item state flag
230     "title",  # Item title (on server is available else local)
231     "size",  # Item size in bytes (on server if available else local)
232     "latest",  # Latest item date (on server), can be None
233     "local",  # Local item date, can be None
234     "tags",  # Item tags (on server if available else local)
235     "info_local",
236     "info_server"]
237)
238
239ItemInfo = namedtuple(
240    "ItemInfo",
241    ["domain",
242     "filename",
243     "title",
244     "time",  # datetime.datetime
245     "size",  # size in bytes
246     "tags"]
247)
248
249
250def UpdateItem_match(item, string):
251    """
252    Return `True` if the `UpdateItem` item contains a string in tags
253    or in the title.
254
255    """
256    string = string.lower()
257    return any(string.lower() in tag.lower()
258               for tag in item.tags + [item.title])
259
260
261def item_state(info_local, info_server):
262    """
263    Return the item state (AVAILABLE, ...) based on it's local and server side
264    `ItemInfo` instances.
265
266    """
267    if info_server is None:
268        return DEPRECATED
269
270    if info_local is None:
271        return AVAILABLE
272
273    if info_local.time < info_server.time:
274        return OUTDATED
275    else:
276        return CURRENT
277
278
279DATE_FMT_1 = "%Y-%m-%d %H:%M:%S.%f"
280DATE_FMT_2 = "%Y-%m-%d %H:%M:%S"
281
282
283def info_dict_to_item_info(domain, filename, item_dict):
284    """
285    Return an `ItemInfo` instance based on `item_dict` as returned by
286    ``serverfiles.info(domain, filename)``
287
288    """
289    time = item_dict["datetime"]
290    try:
291        time = datetime.strptime(time, DATE_FMT_1)
292    except ValueError:
293        time = datetime.strptime(time, DATE_FMT_2)
294
295    title = item_dict["title"]
296    if not title:
297        title = filename
298
299    size = int(item_dict["size"])
300    tags = item_dict["tags"]
301    return ItemInfo(domain, filename, title, time, size, tags)
302
303
304def update_item_from_info(domain, filename, info_server, info_local):
305    """
306    Return a `UpdateItem` instance for `domain`, `fileanme` based on
307    the local and server side `ItemInfo` instances `info_server` and
308    `info_local`.
309
310    """
311    latest, local, title, tags, size = None, None, None, None, None
312
313    if info_server is not None:
314        info_server = info_dict_to_item_info(domain, filename, info_server)
315        latest = info_server.time
316        tags = info_server.tags
317        title = info_server.title
318        size = info_server.size
319
320    if info_local is not None:
321        info_local = info_dict_to_item_info(domain, filename, info_local)
322        local = info_local.time
323
324        if info_server is None:
325            tags = info_local.tags
326            title = info_local.title
327            size = info_local.size
328
329    state = item_state(info_local, info_server)
330
331    return UpdateItem(domain, filename, state, title, size, latest, local,
332                      tags, info_server, info_local)
333
334
335def join_info_list(domain, files_local, files_server):
336    filenames = set(files_local.keys()).union(files_server.keys())
337    for filename in sorted(filenames):
338        info_server = files_server.get(filename, None)
339        info_local = files_local.get(filename, None)
340        yield update_item_from_info(domain, filename, info_server, info_local)
341
342
343def join_info_dict(local, server):
344    domains = set(local.keys()).union(server.keys())
345    for domain in sorted(domains):
346        files_local = local.get(domain, {})
347        files_server = server.get(domain, {})
348
349        for item in join_info_list(domain, files_local, files_server):
350            yield item
351
352
353def special_tags(item):
354    """
355    Return a dictionary of special tags in an UpdateItem instance (special
356    tags are the ones starting with #).
357
358    """
359    return dict([tuple(tag.split(":")) for tag in item.tags
360                 if tag.startswith("#") and ":" in tag])
361
362
363def retrieveFilesList(serverFiles, domains=None, advance=lambda: None):
364    """
365    Retrieve and return serverfiles.allinfo for all domains.
366    """
367    domains = serverFiles.listdomains() if domains is None else domains
368    advance()
369    serverInfo = {}
370    for dom in domains:
371        try:
372            serverInfo[dom] = serverFiles.allinfo(dom)
373        except Exception: #ignore inexistent domains
374            pass
375    advance()
376    return serverInfo
377
378
379class OWDatabasesUpdate(OWWidget):
380    def __init__(self, parent=None, signalManager=None,
381                 name="Databases update", wantCloseButton=False,
382                 searchString="", showAll=True, domains=None,
383                 accessCode=""):
384        OWWidget.__init__(self, parent, signalManager, name, wantMainArea=False)
385        self.searchString = searchString
386        self.accessCode = accessCode
387        self.showAll = showAll
388        self.domains = domains
389        self.serverFiles = serverfiles.ServerFiles()
390
391        box = OWGUI.widgetBox(self.controlArea, orientation="horizontal")
392
393        self.lineEditFilter = \
394            OWGUIEx.lineEditHint(box, self, "searchString", "Filter",
395                                 caseSensitive=False,
396                                 delimiters=" ",
397                                 matchAnywhere=True,
398                                 listUpdateCallback=self.SearchUpdate,
399                                 callbackOnType=True,
400                                 callback=self.SearchUpdate)
401
402        box = OWGUI.widgetBox(self.controlArea, "Files")
403        self.filesView = QTreeWidget(self)
404        self.filesView.setHeaderLabels(
405            ["", "Data Source", "Update", "Last Updated", "Size"])
406
407        self.filesView.setRootIsDecorated(False)
408        self.filesView.setUniformRowHeights(True)
409        self.filesView.setSelectionMode(QAbstractItemView.NoSelection)
410        self.filesView.setSortingEnabled(True)
411        self.filesView.sortItems(1, Qt.AscendingOrder)
412        self.filesView.setItemDelegateForColumn(
413            0, UpdateOptionsItemDelegate(self.filesView))
414
415        QObject.connect(self.filesView.model(),
416                        SIGNAL("layoutChanged()"),
417                        self.SearchUpdate)
418        box.layout().addWidget(self.filesView)
419
420        box = OWGUI.widgetBox(self.controlArea, orientation="horizontal")
421        self.updateButton = OWGUI.button(box, self, "Update all",
422                     callback=self.UpdateAll,
423                     tooltip="Update all updatable files",
424                     )
425       
426        self.downloadButton = OWGUI.button(box, self, "Download all",
427                     callback=self.DownloadFiltered,
428                     tooltip="Download all filtered files shown")
429        self.cancelButton = OWGUI.button(box, self, "Cancel", callback=self.Cancel,
430                     tooltip="Cancel scheduled downloads/updates.")
431        OWGUI.rubber(box)
432        OWGUI.lineEdit(box, self, "accessCode", "Access Code",
433                       orientation="horizontal",
434                       callback=self.RetrieveFilesList)
435        self.retryButton = OWGUI.button(box, self, "Retry",
436                                        callback=self.RetrieveFilesList)
437        self.retryButton.hide()
438        box = OWGUI.widgetBox(self.controlArea, orientation="horizontal")
439        OWGUI.rubber(box)
440        if wantCloseButton:
441            OWGUI.button(box, self, "Close",
442                         callback=self.accept,
443                         tooltip="Close")
444
445        self.infoLabel = QLabel()
446        self.infoLabel.setAlignment(Qt.AlignCenter)
447
448        self.controlArea.layout().addWidget(self.infoLabel)
449        self.infoLabel.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
450
451        self.updateItems = []
452
453        self.resize(800, 600)
454
455        self.progress = ProgressState(self, maximum=3)
456        self.progress.valueChanged.connect(self._updateProgress)
457        self.progress.rangeChanged.connect(self._updateProgress)
458        self.executor = ThreadExecutor(
459            threadPool=QThreadPool(maxThreadCount=2)
460        )
461
462        task = Task(self, function=self.RetrieveFilesList)
463        task.exceptionReady.connect(self.HandleError)
464        task.start()
465
466        self._tasks = []
467        self._haveProgress = False
468
469    def RetrieveFilesList(self):
470        self.progress.setRange(0, 3)
471        self.serverFiles = serverfiles.ServerFiles(access_code=self.accessCode)
472
473        task = Task(function=partial(retrieveFilesList, self.serverFiles,
474                                     self.domains,
475                                     methodinvoke(self.progress, "advance")))
476
477        task.resultReady.connect(self.SetFilesList)
478        task.exceptionReady.connect(self.HandleError)
479
480        self.executor.submit(task)
481
482        self.setEnabled(False)
483
484    def SetFilesList(self, serverInfo):
485        """
486        Set the files to show.
487        """
488        self.setEnabled(True)
489
490        domains = serverInfo.keys()
491        if not domains:
492            if self.domains:
493                domains = self.domains
494            else:
495                domains = serverfiles.listdomains()
496
497        localInfo = dict([(dom, serverfiles.allinfo(dom)) for dom in domains])
498
499        all_tags = set()
500
501        self.filesView.clear()
502        self.updateItems = []
503
504        for item in join_info_dict(localInfo, serverInfo):
505            tree_item = UpdateTreeWidgetItem(item)
506            options_widget = UpdateOptionsWidget(item.state)
507            options_widget.item = item
508
509            options_widget.installClicked.connect(
510                partial(self.SubmitDownloadTask, item.domain, item.filename)
511            )
512            options_widget.removeClicked.connect(
513                partial(self.SubmitRemoveTask, item.domain, item.filename)
514            )
515
516            self.updateItems.append((item, tree_item, options_widget))
517            all_tags.update(item.tags)
518
519        self.filesView.addTopLevelItems(
520            [tree_item for _, tree_item, _ in self.updateItems]
521        )
522
523        for item, tree_item, options_widget in self.updateItems:
524            self.filesView.setItemWidget(tree_item, 0, options_widget)
525
526            # Add an update button if the file is updateable
527            if item.state == OUTDATED:
528                button = QToolButton(
529                    None, text="Update",
530                    maximumWidth=120,
531                    maximumHeight=30
532                )
533
534                if sys.platform == "darwin":
535                    button.setAttribute(Qt.WA_MacSmallSize)
536
537                button.clicked.connect(
538                    partial(self.SubmitDownloadTask, item.domain,
539                            item.filename)
540                )
541
542                self.filesView.setItemWidget(tree_item, 2, button)
543
544        self.progress.advance()
545
546        self.filesView.setColumnWidth(0, self.filesView.sizeHintForColumn(0))
547
548        for column in range(1, 4):
549            contents_hint = self.filesView.sizeHintForColumn(column)
550            header_hint = self.filesView.header().sectionSizeHint(column)
551            width = max(min(contents_hint, 400), header_hint)
552            self.filesView.setColumnWidth(column, width)
553
554        self.lineEditFilter.setItems([hint for hint in sorted(all_tags)
555                                      if not hint.startswith("#")])
556        self.SearchUpdate()
557        self.UpdateInfoLabel()
558        self.toggleButtons()
559        self.cancelButton.setEnabled(False)
560
561        self.progress.setRange(0, 0)
562
563    def buttonCheck(self, selected_items, state, button):
564        for item in selected_items:
565            if item.state != state:
566                button.setEnabled(False)
567            else:
568                button.setEnabled(True)
569                break
570
571    def toggleButtons(self):
572        selected_items = [item for item, tree_item, _ in self.updateItems if not tree_item.isHidden()]
573        self.buttonCheck(selected_items, OUTDATED, self.updateButton)
574        self.buttonCheck(selected_items, AVAILABLE, self.downloadButton)
575
576    def HandleError(self, exception):
577        if isinstance(exception, IOError):
578            self.error(0,
579                       "Could not connect to server! Press the Retry "
580                       "button to try again.")
581            self.SetFilesList({})
582        else:
583            sys.excepthook(type(exception), exception.args, None)
584            self.progress.setRange(0, 0)
585            self.setEnabled(True)
586
587    def UpdateInfoLabel(self):
588        local = [item for item, tree_item, _ in self.updateItems
589                 if item.state != AVAILABLE and not tree_item.isHidden() ]
590        size = sum(float(item.size) for item in local)
591
592        onServer = [item for item, tree_item, _ in self.updateItems if not tree_item.isHidden()]
593        sizeOnServer = sum(float(item.size) for item in onServer)
594
595        text = ("%i items, %s (on server: %i items, %s)" %
596                (len(local),
597                 sizeof_fmt(size),
598                 len(onServer),
599                 sizeof_fmt(sizeOnServer)))
600
601        self.infoLabel.setText(text)
602
603    def UpdateAll(self):
604        for item, tree_item, _ in self.updateItems:
605            if item.state == OUTDATED and not tree_item.isHidden():
606                self.SubmitDownloadTask(item.domain, item.filename)
607
608    def DownloadFiltered(self):
609        # TODO: submit items in the order shown.
610        for item, tree_item, _ in self.updateItems:
611            if not tree_item.isHidden() and item.state in \
612                    [AVAILABLE, OUTDATED]:
613                self.SubmitDownloadTask(item.domain, item.filename)
614
615    def SearchUpdate(self, searchString=None):
616        strings = unicode(self.lineEditFilter.text()).split()
617        for item, tree_item, _ in self.updateItems:
618            hide = not all(UpdateItem_match(item, string)
619                           for string in strings)
620            tree_item.setHidden(hide)
621        self.UpdateInfoLabel()
622        self.toggleButtons()
623
624    def SubmitDownloadTask(self, domain, filename):
625        """
626        Submit the (domain, filename) to be downloaded/updated.
627        """
628        self.cancelButton.setEnabled(True)
629
630        index = self.updateItemIndex(domain, filename)
631        _, tree_item, opt_widget = self.updateItems[index]
632
633        if self.accessCode:
634            sf = serverfiles.ServerFiles(access_code=self.accessCode)
635        else:
636            sf = serverfiles.ServerFiles()
637
638        task = DownloadTask(domain, filename, sf)
639
640        self.executor.submit(task)
641
642        self.progress.adjustRange(0, 100)
643
644        pb = ItemProgressBar(self.filesView)
645        pb.setRange(0, 100)
646        pb.setTextVisible(False)
647
648        task.advanced.connect(pb.advance)
649        task.advanced.connect(self.progress.advance)
650        task.finished.connect(pb.hide)
651        task.finished.connect(self.onDownloadFinished, Qt.QueuedConnection)
652        task.exception.connect(self.onDownloadError, Qt.QueuedConnection)
653
654        self.filesView.setItemWidget(tree_item, 2, pb)
655
656        # Clear the text so it does not show behind the progress bar.
657        tree_item.setData(2, Qt.DisplayRole, "")
658        pb.show()
659
660        # Disable the options widget
661        opt_widget.setEnabled(False)
662        self._tasks.append(task)
663
664    def EndDownloadTask(self, task):
665        future = task.future()
666        index = self.updateItemIndex(task.domain, task.filename)
667        item, tree_item, opt_widget = self.updateItems[index]
668
669        self.filesView.removeItemWidget(tree_item, 2)
670        opt_widget.setEnabled(True)
671
672        if future.cancelled():
673            # Restore the previous state
674            tree_item.setUpdateItem(item)
675            opt_widget.setState(item.state)
676
677        elif future.exception():
678            tree_item.setUpdateItem(item)
679            opt_widget.setState(item.state)
680
681            # Show the exception string in the size column.
682            tree_item.setData(2, Qt.DisplayRole,
683                         QVariant("Error occurred while downloading:" +
684                                  str(future.exception())))
685
686        else:
687            # get the new updated info dict and replace the the old item
688            info = serverfiles.info(item.domain, item.filename)
689            new_item = update_item_from_info(item.domain, item.filename,
690                                             info, info)
691
692            self.updateItems[index] = (new_item, tree_item, opt_widget)
693
694            tree_item.setUpdateItem(new_item)
695            opt_widget.setState(new_item.state)
696
697            self.UpdateInfoLabel()
698            self.cancelButton.setEnabled(False)
699
700    def SubmitRemoveTask(self, domain, filename):
701        serverfiles.remove(domain, filename)
702        index = self.updateItemIndex(domain, filename)
703        item, tree_item, opt_widget = self.updateItems[index]
704
705        if item.info_server:
706            new_item = item._replace(state=AVAILABLE, local=None,
707                                      info_local=None)
708        else:
709            new_item = item._replace(local=None, info_local=None)
710            # Disable the options widget. No more actions can be performed
711            # for the item.
712            opt_widget.setEnabled(False)
713
714        tree_item.setUpdateItem(new_item)
715        opt_widget.setState(new_item.state)
716        self.updateItems[index] = (new_item, tree_item, opt_widget)
717
718        self.UpdateInfoLabel()
719
720    def Cancel(self):
721        """
722        Cancel all pending update/download tasks (that have not yet started).
723        """
724        for task in self._tasks:
725            task.future().cancel()
726
727    def onDeleteWidget(self):
728        self.Cancel()
729        self.executor.shutdown(wait=False)
730        OWBaseWidget.onDeleteWidget(self)
731
732    def onDownloadFinished(self):
733        assert QThread.currentThread() is self.thread()
734        for task in list(self._tasks):
735            future = task.future()
736            if future.done():
737                self.EndDownloadTask(task)
738                self._tasks.remove(task)
739
740        if not self._tasks:
741            # Clear/reset the overall progress
742            self.progress.setRange(0, 0)
743
744    def onDownloadError(self, exc_info):
745        sys.excepthook(*exc_info)
746
747    def updateItemIndex(self, domain, filename):
748        for i, (item, _, _) in enumerate(self.updateItems):
749            if item.domain == domain and item.filename == filename:
750                return i
751        raise ValueError("%r, %r not in update list" % (domain, filename))
752
753    def _updateProgress(self, *args):
754        rmin, rmax = self.progress.range()
755        if rmin != rmax:
756            if not self._haveProgress:
757                self._haveProgress = True
758                self.progressBarInit()
759
760            self.progressBarSet(self.progress.ratioCompleted() * 100,
761                                processEventsFlags=None)
762        if rmin == rmax:
763            self._haveProgress = False
764            self.progressBarFinished()
765
766
767class ProgressState(QObject):
768    valueChanged = Signal(int)
769    rangeChanged = Signal(int, int)
770    textChanged = Signal(str)
771    started = Signal()
772    finished = Signal()
773
774    def __init__(self, parent=None, minimum=0, maximum=0, text="", value=0):
775        QObject.__init__(self, parent)
776
777        self._minimum = minimum
778        self._maximum = max(maximum, minimum)
779        self._text = text
780        self._value = value
781
782    @Slot(int, int)
783    def setRange(self, minimum, maximum):
784        maximum = max(maximum, minimum)
785
786        if self._minimum != minimum or self._maximum != maximum:
787            self._minimum = minimum
788            self._maximum = maximum
789            self.rangeChanged.emit(minimum, maximum)
790
791            # Adjust the value to fit in the range
792            newvalue = min(max(self._value, minimum), maximum)
793            if newvalue != self._value:
794                self.setValue(newvalue)
795
796    def range(self):
797        return self._minimum, self._maximum
798
799    @Slot(int)
800    def setValue(self, value):
801        if self._value != value and value >= self._minimum and \
802                value <= self._maximum:
803            self._value = value
804            self.valueChanged.emit(value)
805
806    def value(self):
807        return self._value
808
809    @Slot(str)
810    def setText(self, text):
811        if self._text != text:
812            self._text = text
813            self.textChanged.emit(text)
814
815    def text(self):
816        return self._text
817
818    @Slot()
819    @Slot(int)
820    def advance(self, value=1):
821        self.setValue(self._value + value)
822
823    def adjustRange(self, dmin, dmax):
824        self.setRange(self._minimum + dmin, self._maximum + dmax)
825
826    def ratioCompleted(self):
827        span = self._maximum - self._minimum
828        if span < 1e-3:
829            return 0.0
830
831        return min(max(float(self._value - self._minimum) / span, 0.0), 1.0)
832
833
834class DownloadTask(Task):
835    advanced = Signal()
836    exception = Signal(tuple)
837
838    def __init__(self, domain, filename, serverfiles, parent=None):
839        Task.__init__(self, parent)
840        self.filename = filename
841        self.domain = domain
842        self.serverfiles = serverfiles
843        self._interrupt = False
844
845    def interrupt(self):
846        """
847        Interrupt the download.
848        """
849        self._interrupt = True
850
851    def _advance(self):
852        self.advanced.emit()
853        if self._interrupt:
854            raise KeyboardInterrupt
855
856    def run(self):
857        try:
858            serverfiles.download(self.domain, self.filename, self.serverfiles,
859                                 callback=self._advance)
860        except Exception:
861            self.exception.emit(sys.exc_info())
862
863
864if __name__ == "__main__":
865    app = QApplication(sys.argv)
866    w = OWDatabasesUpdate(wantCloseButton=True)
867    w.show()
868    w.exec_()
Note: See TracBrowser for help on using the repository browser.