Mercurial Repositories > eric / file diff

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/eric7/WebBrowser/UserAgent/UserAgentModel.py	Sat May 15 18:45:04 2021 +0200
@@ -0,0 +1,137 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2012 - 2021 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing a model for user agent management.
+"""
+
+import contextlib
+
+from PyQt5.QtCore import Qt, QModelIndex, QAbstractTableModel
+
+
+class UserAgentModel(QAbstractTableModel):
+    """
+    Class implementing a model for user agent management.
+    """
+    def __init__(self, manager, parent=None):
+        """
+        Constructor
+        
+        @param manager reference to the user agent manager (UserAgentManager)
+        @param parent reference to the parent object (QObject)
+        """
+        super().__init__(parent)
+        
+        self.__manager = manager
+        self.__manager.changed.connect(self.__userAgentsChanged)
+        
+        self.__headers = [
+            self.tr("Host"),
+            self.tr("User Agent String"),
+        ]
+    
+    def __userAgentsChanged(self):
+        """
+        Private slot handling a change of the registered user agent strings.
+        """
+        self.beginResetModel()
+        self.endResetModel()
+    
+    def removeRows(self, row, count, parent=None):
+        """
+        Public method to remove entries from the model.
+        
+        @param row start row (integer)
+        @param count number of rows to remove (integer)
+        @param parent parent index (QModelIndex)
+        @return flag indicating success (boolean)
+        """
+        if parent is None:
+            parent = QModelIndex()
+        
+        if parent.isValid():
+            return False
+        
+        if count <= 0:
+            return False
+        
+        lastRow = row + count - 1
+        
+        self.beginRemoveRows(parent, row, lastRow)
+        
+        hostsList = self.__manager.allHostNames()
+        for index in range(row, lastRow + 1):
+            self.__manager.removeUserAgent(hostsList[index])
+        
+        return True
+    
+    def rowCount(self, parent=None):
+        """
+        Public method to get the number of rows of the model.
+        
+        @param parent parent index (QModelIndex)
+        @return number of rows (integer)
+        """
+        if parent is None:
+            parent = QModelIndex()
+        
+        if parent.isValid():
+            return 0
+        else:
+            return self.__manager.hostsCount()
+    
+    def columnCount(self, parent=None):
+        """
+        Public method to get the number of columns of the model.
+        
+        @param parent parent index (QModelIndex) (Unused)
+        @return number of columns (integer)
+        """
+        return len(self.__headers)
+    
+    def data(self, index, role):
+        """
+        Public method to get data from the model.
+        
+        @param index index to get data for (QModelIndex)
+        @param role role of the data to retrieve (integer)
+        @return requested data
+        """
+        if index.row() >= self.__manager.hostsCount() or index.row() < 0:
+            return None
+        
+        host = self.__manager.allHostNames()[index.row()]
+        userAgent = self.__manager.userAgent(host)
+        
+        if userAgent is None:
+            return None
+        
+        if role == Qt.ItemDataRole.DisplayRole:
+            if index.column() == 0:
+                return host
+            elif index.column() == 1:
+                return userAgent
+        
+        return None
+    
+    def headerData(self, section, orientation,
+                   role=Qt.ItemDataRole.DisplayRole):
+        """
+        Public method to get the header data.
+        
+        @param section section number (integer)
+        @param orientation header orientation (Qt.Orientation)
+        @param role data role (Qt.ItemDataRole)
+        @return header data
+        """
+        if (
+            orientation == Qt.Orientation.Horizontal and
+            role == Qt.ItemDataRole.DisplayRole
+        ):
+            with contextlib.suppress(IndexError):
+                return self.__headers[section]
+        
+        return None