eric: comparison WebBrowser/GreaseMonkey/GreaseMonkeyScript.py

-:00c723b11d4d
+:dfe864318196
 from __future__ import unicode_literals
 from PyQt5.QtCore import pyqtSignal, pyqtSlot, QObject, QUrl, QRegExp, \
 QByteArray, QCryptographicHash
+from PyQt5.QtGui import QIcon, QPixmap, QImage
+from PyQt5.QtNetwork import QNetworkRequest, QNetworkReply
 from PyQt5.QtWebEngineWidgets import QWebEngineScript
 from .GreaseMonkeyJavaScript import bootstrap_js, values_js
 from .GreaseMonkeyDownloader import GreaseMonkeyDownloader
 from ..Tools.DelayedFileWatcher import DelayedFileWatcher
 from ..WebBrowserPage import WebBrowserPage
+from ..WebBrowserWindow import WebBrowserWindow
 from Globals import qVersionTuple
 class GreaseMonkeyScript(QObject):
 self.__include = []
 self.__exclude = []
 self.__require = []
+self.__icon = QIcon()
+self.__iconUrl = QUrl()
 self.__downloadUrl = QUrl()
 self.__updateUrl = QUrl()
 self.__startAt = GreaseMonkeyScript.DocumentEnd
 self.__script = ""
 self.__noFrames = False
 self.__updating = False
 self.__downloaders = []
+self.__iconReplies = []
 self.__parseScript()
 self.__fileWatcher.delayedFileChanged.connect(
 self.__watchedFileChanged)
 Public method to get the version of the script.
 @return version of the script (string)
 """
 return self.__version
+def icon(self):
+"""
+Public method to get the icon of the script.
+@return script icon
+@rtype QIcon
+"""
+return self.__icon
+def iconUrl(self):
+"""
+Public method to get the icon URL of the script.
+@return icon URL of the script (QUrl)
+"""
+return QUrl(self.__iconUrl)
 def downloadUrl(self):
 """
 Public method to get the download URL of the script.
 self.__include = []
 self.__exclude = []
 self.__require = []
+self.__icon = QIcon()
+self.__iconUrl = QUrl()
 self.__downloadUrl = QUrl()
 self.__updateUrl = QUrl()
 self.__startAt = GreaseMonkeyScript.DocumentEnd
 self.__script = ""
 if not line.startswith("// @"):
 continue
 line = line[3:].replace("\t", " ")
 index = line.find(" ")
-if index < 0:
-continue
 key = line[:index].strip()
-value = line[index + 1:].strip()
+if index > 0:
+value = line[index + 1:].strip()
-# Ignored values: @resource, @unwrap
+else:
+value = ""
-if not key or not value:
+if not key:
 continue
 if key == "@name":
 self.__name = value
 elif key == "@downloadURL" and self.__downloadUrl.isEmpty():
 self.__downloadUrl = QUrl(value)
 elif key == "@updateURL" and self.__updateUrl.isEmpty():
 self.__updateUrl = QUrl(value)
+elif key == "@icon":
+self.__iconUrl = QUrl(value)
+elif key == "@noframes":
+self.__noFrames = True
+self.__iconUrl = self.__downloadUrl.resolved(self.__iconUrl)
 if not self.__include:
 self.__include.append("*")
 nspace = bytes(QCryptographicHash.hash(
 self.__script = "(function(){{{0}\n{1}\n{2}\n}})();".format(
 valuesScript, self.__manager.requireScripts(self.__require),
 fileData
 )
 self.__valid = True
+self.__downloadIcon()
+self.__downloadRequires()
 def webScript(self):
 """
 Public method to create a script object.
 @param downloader reference to the downloader object
 @type GreaseMonkeyDownloader
 """
 if downloader in self.__downloaders:
 self.__downloaders.remove(downloader)
+def __downloadIcon(self):
+"""
+Private slot to download the script icon.
+"""
+if self.__iconUrl.isValid():
+request = QNetworkRequest(self.__iconUrl)
+reply = WebBrowserWindow.networkManager().get(request)
+reply.finished.connect(lambda: self.__iconDownloaded(reply))
+self.__iconReplies.append(reply)
+def __iconDownloaded(self, reply):
+"""
+Private slot to handle a finished download of a script icon.
+@param reply reference to the network reply
+@type QNetworkReply
+"""
+if reply in self.__iconReplies:
+self.__iconReplies.remove(reply)
+reply.deleteLater()
+if reply.error() == QNetworkReply.NoError:
+self.__icon = QPixmap.fromImage(QImage.fromData(reply.readAll()))

Mercurial Repositories > eric / file comparison

comparison: WebBrowser/GreaseMonkey/GreaseMonkeyScript.py

WebBrowser/GreaseMonkey/GreaseMonkeyScript.py