Mercurial Repositories > eric / file diff

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/eric7/WebBrowser/FeaturePermissions/FeaturePermissionManager.py	Sat May 15 18:45:04 2021 +0200
@@ -0,0 +1,231 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2015 - 2021 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing the feature permission manager object.
+"""
+
+import contextlib
+
+from PyQt5.QtCore import QObject
+from PyQt5.QtWidgets import QDialog
+from PyQt5.QtWebEngineWidgets import QWebEnginePage
+
+import Globals
+import Preferences
+
+
+class FeaturePermissionManager(QObject):
+    """
+    Class implementing the feature permission manager object.
+    """
+    SettingsKeyFormat = "WebBrowser/FeaturePermissions/{0}"
+    
+    def __init__(self, parent=None):
+        """
+        Constructor
+        
+        @param parent reference to the parent object
+        @type QObject
+        """
+        super().__init__(parent)
+        
+        self.__featurePermissions = {
+            QWebEnginePage.Feature.Geolocation: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser: [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser: [],
+            },
+            QWebEnginePage.Feature.MediaAudioCapture: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser: [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser: [],
+            },
+            QWebEnginePage.Feature.MediaVideoCapture: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser: [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser: [],
+            },
+            QWebEnginePage.Feature.MediaAudioVideoCapture: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser: [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser: [],
+            },
+            QWebEnginePage.Feature.MouseLock: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser: [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser: [],
+            },
+            QWebEnginePage.Feature.DesktopVideoCapture: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser:
+                    [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser:
+                    [],
+            },
+            QWebEnginePage.Feature.DesktopAudioVideoCapture: {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser:
+                    [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser:
+                    [],
+            },
+        }
+        with contextlib.suppress(AttributeError):
+            # this was re-added in Qt 5.13.0
+            self.__featurePermissions[QWebEnginePage.Feature.Notifications] = {
+                QWebEnginePage.PermissionPolicy.PermissionGrantedByUser: [],
+                QWebEnginePage.PermissionPolicy.PermissionDeniedByUser: [],
+            }
+        
+        self.__featurePermissionsKeys = {
+            (QWebEnginePage.Feature.Geolocation,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "GeolocationGranted",
+            (QWebEnginePage.Feature.Geolocation,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "GeolocationDenied",
+            (QWebEnginePage.Feature.MediaAudioCapture,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "MediaAudioCaptureGranted",
+            (QWebEnginePage.Feature.MediaAudioCapture,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "MediaAudioCaptureDenied",
+            (QWebEnginePage.Feature.MediaVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "MediaVideoCaptureGranted",
+            (QWebEnginePage.Feature.MediaVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "MediaVideoCaptureDenied",
+            (QWebEnginePage.Feature.MediaAudioVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "MediaAudioVideoCaptureGranted",
+            (QWebEnginePage.Feature.MediaAudioVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "MediaAudioVideoCaptureDenied",
+            (QWebEnginePage.Feature.MouseLock,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "MouseLockGranted",
+            (QWebEnginePage.Feature.MouseLock,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "MouseLockDenied",
+            (QWebEnginePage.Feature.DesktopVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "DesktopVideoCaptureGranted",
+            (QWebEnginePage.Feature.DesktopVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "DesktopVideoCaptureDenied",
+            (QWebEnginePage.Feature.DesktopAudioVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+            "DesktopAudioVideoCaptureGranted",
+            (QWebEnginePage.Feature.DesktopAudioVideoCapture,
+             QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+            "DesktopAudioVideoCaptureDenied",
+        }
+        with contextlib.suppress(AttributeError):
+            # this was re-added in Qt 5.13.0
+            self.__featurePermissionsKeys.update({
+                (QWebEnginePage.Feature.Notifications,
+                 QWebEnginePage.PermissionPolicy.PermissionGrantedByUser):
+                "NotificationsGranted",
+                (QWebEnginePage.Feature.Notifications,
+                 QWebEnginePage.PermissionPolicy.PermissionDeniedByUser):
+                "NotificationsDenied",
+            })
+        
+        self.__loaded = False
+
+    def requestFeaturePermission(self, page, origin, feature):
+        """
+        Public method to request a feature permission.
+        
+        @param page reference to the requesting web page
+        @type QWebEnginePage
+        @param origin security origin requesting the feature
+        @type QUrl
+        @param feature requested feature
+        @type QWebEnginePage.Feature
+        """
+        if origin is None or origin.isEmpty():
+            return
+        
+        if not self.__loaded:
+            self.__loadSettings()
+        
+        host = origin.host()
+        
+        if feature in self.__featurePermissions:
+            for permission in self.__featurePermissions[feature]:
+                if host in self.__featurePermissions[feature][permission]:
+                    page.setFeaturePermission(origin, feature, permission)
+                    return
+        
+        from .FeaturePermissionBar import FeaturePermissionBar
+        bar = FeaturePermissionBar(page, origin, feature, self)
+        bar.show()
+    
+    def rememberFeaturePermission(self, host, feature, permission):
+        """
+        Public method to remember a user decision for a feature permission.
+        
+        @param host host name to remember the decision for
+        @type str
+        @param feature feature to be remembered
+        @type QWebEnginePage.Feature
+        @param permission feature permission to be remembered
+        @type QWebEnginePage.PermissionPolicy
+        """
+        if (
+            feature in self.__featurePermissions and
+            host not in self.__featurePermissions[feature][permission]
+        ):
+            self.__featurePermissions[feature][permission].append(host)
+            self.__saveSettings()
+    
+    def __loadSettings(self):
+        """
+        Private method to load the remembered feature permissions.
+        """
+        if self.__loaded:
+            # no reloading allowed
+            return
+        
+        for (feature, permission), key in (
+            self.__featurePermissionsKeys.items()
+        ):
+            self.__featurePermissions[feature][permission] = (
+                Globals.toList(Preferences.Prefs.settings.value(
+                    FeaturePermissionManager.SettingsKeyFormat.format(key),
+                    []
+                ))
+            )
+        
+        self.__loaded = True
+    
+    def __saveSettings(self):
+        """
+        Private method to save the remembered feature permissions.
+        """
+        if not self.__loaded:
+            return
+        
+        import WebBrowser.WebBrowserWindow
+        if WebBrowser.WebBrowserWindow.WebBrowserWindow.isPrivate():
+            return
+        
+        for (feature, permission), key in (
+                self.__featurePermissionsKeys.items()
+        ):
+            Preferences.Prefs.settings.setValue(
+                FeaturePermissionManager.SettingsKeyFormat.format(key),
+                self.__featurePermissions[feature][permission])
+    
+    def showFeaturePermissionsDialog(self):
+        """
+        Public method to show a dialog to manage the remembered feature
+        permissions.
+        """
+        if not self.__loaded:
+            self.__loadSettings()
+        
+        from .FeaturePermissionsDialog import FeaturePermissionsDialog
+        dlg = FeaturePermissionsDialog(self.__featurePermissions)
+        if dlg.exec() == QDialog.DialogCode.Accepted:
+            newFeaturePermissions = dlg.getData()
+            self.__featurePermissions = newFeaturePermissions
+            self.__saveSettings()