diff -r 7a523bd4b00d -r 11f8adbcac97 Helpviewer/Download/DownloadAskActionDialog.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Helpviewer/Download/DownloadAskActionDialog.py Sun Apr 10 18:33:20 2011 +0200 @@ -0,0 +1,51 @@ +# -*- coding: utf-8 -*- + +# Copyright (c) 2011 Detlev Offenbach <detlev@die-offenbachs.de> +# + +""" +Module implementing a dialog to ask for a download action. +""" + +from PyQt4.QtGui import QDialog + +from .Ui_DownloadAskActionDialog import Ui_DownloadAskActionDialog + +import Preferences + +class DownloadAskActionDialog(QDialog, Ui_DownloadAskActionDialog): + """ + Class implementing a dialog to ask for a download action. + """ + def __init__(self, fileName, mimeType, baseUrl, parent=None): + """ + Constructor + + @param parent reference to the parent widget (QWidget) + """ + QDialog.__init__(self, parent) + self.setupUi(self) + + self.infoLabel.setText("<b>{0}</b>".format(fileName)) + self.typeLabel.setText(mimeType) + self.siteLabel.setText(baseUrl) + + if not Preferences.getHelp("VirusTotalEnabled") or \ + Preferences.getHelp("VirusTotalServiceKey") == "": + self.scanButton.setHidden(True) + + def getAction(self): + """ + Public method to get the selected action. + + @return selected action ("save", "open", "scan" or "cancel") + """ + if self.openButton.isChecked(): + return "open" + elif self.scanButton.isChecked(): + return "scan" + elif self.saveButton.isChecked(): + return "save" + else: + # should not happen, but keep it safe + return "cancel"