123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239 |
- from PyQt5.QtWidgets import (
- QWidget,
- QVBoxLayout,
- QLabel,
- QTableView,
- QHeaderView,
- QAbstractItemView,
- QHBoxLayout,
- QFormLayout,
- QCheckBox,
- QMessageBox,
- QMenu,
- QComboBox,
- QSizePolicy,
- QScrollArea,
- QSplitter,
- QFrame,
- QWidgetAction
- )
- from PyQt5.QtCore import pyqtSignal, Qt, pyqtProperty
- from PyQt5.QtGui import QGuiApplication, QPainter, QFontMetrics, QColor
- import json
- from searxqt.core.handler import NetworkTypes
- from searxqt.core.requests import ErrorType
- from searxqt.widgets.buttons import Button
- from searxqt.widgets.dialogs import UrlDialog, RequestsErrorMessage
- from searxqt.models.instances import (
- InstanceTableModel,
- InstancesModelTypes
- )
- from searxqt.translations import _, timeToString
- class AddUserInstanceDialog(UrlDialog):
- def __init__(self, url='', parent=None):
- UrlDialog.__init__(self, url=url, acceptTxt=_("Add"), parent=parent)
- layout = self.layout()
- label = QLabel("Update data on add:")
- self._updateCheckBox = QCheckBox("", self)
- layout.insertRow(2, label, self._updateCheckBox)
- def updateOnAdd(self):
- return bool(self._updateCheckBox.isChecked())
- class InstancesView(QWidget):
- def __init__(self, filterModel, instanceSelecter, parent=None):
- """
- @type model: searxqt.models.instances.InstanceModelFilter
- @type instanceSelecter: searxqt.models.instances.InstanceSelecterModel
- @type parent: QObject
- """
- QWidget.__init__(self, parent=parent)
- layout = QVBoxLayout(self)
- layout.setContentsMargins(0, 0, 0, 0)
- layout.setSpacing(0)
- self._filterModel = filterModel
- self._instanceSelecter = instanceSelecter
- self._tableModel = InstanceTableModel(filterModel, self)
-
- self._currentInstancesType = filterModel.parentModel().Type
- headLabel = QLabel(f"<h2>{_('Instances')}</h2>", self)
- layout.addWidget(headLabel)
-
- self._splitter = QSplitter(self)
- self._splitter.setOrientation(Qt.Vertical)
- layout.addWidget(self._splitter)
-
- self._scrollArea = QScrollArea(self._splitter)
- self._scrollContentWidget = QWidget(self._scrollArea)
- scrollLayout = QVBoxLayout(self._scrollContentWidget)
- self.filterWidget = FilterWidget(
- filterModel,
- self._scrollContentWidget
- )
- self._scrollArea.setWidget(self._scrollContentWidget)
- self._scrollArea.setFrameShape(QFrame.NoFrame)
- self._scrollArea.setWidgetResizable(True)
- self._scrollArea.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
- self._scrollArea.setAlignment(
- Qt.AlignLeading | Qt.AlignLeft | Qt.AlignTop
- )
- scrollLayout.addWidget(self.filterWidget, 0, Qt.AlignTop)
-
- bottomWidget = QWidget(self._splitter)
- bottomLayout = QVBoxLayout(bottomWidget)
- self.statsWidget = InstancesStatsWidget(filterModel, bottomWidget)
- bottomLayout.addWidget(self.statsWidget)
- self._typeDependantButton = Button("", bottomWidget)
- bottomLayout.addWidget(self._typeDependantButton)
- self._tableView = InstanceTableView(
- filterModel,
- self._tableModel,
- self
- )
- bottomLayout.addWidget(self._tableView)
-
- self._tableView.selectionModel().selectionChanged.connect(
- self.__selectionChanged)
- self._tableModel.layoutChanged.connect(self.__tableModelLayoutChanged)
- self._instanceSelecter.instanceChanged.connect(self.__instanceChanged)
- filterModel.parentModel().typeChanged.connect(
- self.__instancesModelTypeChanged
- )
- """ InstancesType dependant methods below
- """
- def __instancesModelTypeChanged(self, type_):
- """ PersistentInstancesModel emitted changed.
- """
- previousType = self._currentInstancesType
-
- if previousType == InstancesModelTypes.Stats2:
- self._typeDependantButton.clicked.disconnect(self.__updateClicked)
- elif previousType == InstancesModelTypes.User:
- self._typeDependantButton.clicked.disconnect(self.__addClicked)
- if type_ == InstancesModelTypes.Stats2:
- self._typeDependantButton.setText(_("Update"))
- self._typeDependantButton.clicked.connect(self.__updateClicked)
- elif type_ == InstancesModelTypes.User:
- self._typeDependantButton.setText(_("Add Instance"))
- self._typeDependantButton.clicked.connect(self.__addClicked)
- self._currentInstancesType = type_
-
- def __updateClicked(self, checked):
- self._typeDependantButton.setEnabled(False)
- handler = self._filterModel.parentModel().handler()
- if handler.updateInstances():
- handler.updateFinished.connect(self.__updateFinished)
- def __updateFinished(self, errorType, errorMsg):
- handler = self._filterModel.parentModel().handler()
- handler.updateFinished.disconnect(self.__updateFinished)
- if errorType != ErrorType.Success:
- dialog = RequestsErrorMessage(errorType, errorMsg)
- dialog.exec()
- self._typeDependantButton.setEnabled(True)
-
- def __addClicked(self, checked):
- dialog = AddUserInstanceDialog()
- if dialog.exec():
- handler = self._filterModel.parentModel().handler()
- handler.addInstance(dialog.url)
- if dialog.updateOnAdd():
- handler.updateInstance(dialog.url)
- """ -----------------------------------------------------------
- """
- def selectUrl(self, url):
- self._tableView.selectionModel().selectionChanged.disconnect(
- self.__selectionChanged)
-
- horScrollVal = self._tableView.horizontalScrollBar().value()
- if url in self._filterModel:
- rowIndex = self._tableModel.getByUrl(url)
- modelIndex = self._tableModel.index(rowIndex, 0)
- self._tableView.setCurrentIndex(modelIndex)
- self._tableView.scrollTo(
- modelIndex,
- QAbstractItemView.PositionAtCenter
- )
- else:
-
- self._tableView.clearSelection()
-
- self._tableView.horizontalScrollBar().setValue(horScrollVal)
- self._tableView.selectionModel().selectionChanged.connect(
- self.__selectionChanged)
- def __instanceChanged(self, url):
- """ Select and scroll to table item with passed url.
- @param url: Instance url to select and scroll to.
- @type url: str
- """
- if url in self._filterModel:
- rowIndex = self._tableModel.getByUrl(url)
- modelIndex = self._tableModel.index(rowIndex, 0)
- self._tableView.setCurrentIndex(modelIndex)
- self._tableView.scrollTo(
- modelIndex,
- QAbstractItemView.PositionAtCenter
- )
- def __tableModelLayoutChanged(self):
- """ Re-select the current URL when the layout changed.
- """
- self.selectUrl(self._instanceSelecter.currentUrl)
- def __selectionChanged(self):
- """ Let the InstanceSelecter know that the current instance has
- changed. (Selected by the user.)
- """
- self._instanceSelecter.instanceChanged.disconnect(
- self.__instanceChanged)
- selectionModel = self._tableView.selectionModel()
- if selectionModel.hasSelection():
- rows = selectionModel.selectedRows()
- selectedIndex = rows[0].row()
- self._instanceSelecter.currentUrl = (
- self._tableModel.getByIndex(selectedIndex)
- )
- else:
- self._instanceSelecter.currentUrl = ""
- self._instanceSelecter.instanceChanged.connect(self.__instanceChanged)
- def saveSettings(self):
- header = self._tableView.horizontalHeader()
- data = {
- 'hiddenColumnIndexes': self._tableView.getHiddenColumnIndexes(),
- 'headerState': header.saveState(),
- 'splitterState': self._splitter.saveState()
- }
- return data
- def loadSettings(self, data):
- self._tableView.setHiddenColumnIndexes(
- data.get('hiddenColumnIndexes', [])
- )
- if 'headerState' in data:
- header = self._tableView.horizontalHeader()
- header.restoreState(data['headerState'])
- if 'splitterState' in data:
- self._splitter.restoreState(data['splitterState'])
- class InstanceTableView(QTableView):
- def __init__(self, filterModel, tableModel=None, parent=None):
- """
- @type filterModel: searxqt.models.instances.InstanceModelFilter
- @type tableModel: searxqt.models.instances.InstanceTableModel
- """
- QTableView.__init__(self, parent=parent)
- self._filterModel = filterModel
- if tableModel:
- self.setModel(tableModel)
- self.setAlternatingRowColors(True)
- self.setSelectionBehavior(QAbstractItemView.SelectRows)
- self.setEditTriggers(QAbstractItemView.NoEditTriggers)
- self.setSortingEnabled(True)
- self.setHorizontalScrollMode(QAbstractItemView.ScrollPerPixel)
-
- header = self.horizontalHeader()
- header.setSectionResizeMode(QHeaderView.ResizeToContents)
- header.setSectionsMovable(True)
- def _hideColumn(self, index, state):
- state = state == 0
- self.setColumnHidden(index, state)
- def getHiddenColumnIndexes(self):
- columnIndex = 0
- indexes = []
- for columnData in self.model().getColumns():
-
- if self.isColumnHidden(columnIndex):
- indexes.append(columnIndex)
- columnIndex += 1
- return indexes
- def setHiddenColumnIndexes(self, indexes):
- for index in indexes:
- self._hideColumn(index, 0)
- """ Re-implementations
- """
- def contextMenuEvent(self, event):
- def __removeFromWhitelist(self, url):
-
- self._filterModel.delInstanceFromWhitelist(url)
- selectionModel = self.selectionModel()
- if selectionModel.hasSelection():
- menu = QMenu(self)
-
- addToBlacklistAction = menu.addAction(_("Add to blacklist"))
-
- addToTmpBlacklistAction = menu.addAction(_("Temporary blacklist"))
-
- addToWhitelistAction = menu.addAction(_("Add to whitelist"))
- menu.addSeparator()
-
- copyMenu = QMenu(_("Copy column text"), self)
- copyMenuActions = []
- for columnData in self.model().getColumns():
- action = copyMenu.addAction(columnData.name)
- copyMenuActions.append(action)
- menu.addMenu(copyMenu)
-
- copyJson = menu.addAction(_("Copy JSON data"))
- menu.addSeparator()
-
- selectAllAction = menu.addAction(_("Select All"))
- menu.addSeparator()
-
- columnsMenu = QMenu(_("Columns"), self)
- menu.addMenu(columnsMenu)
-
- userInstances = False
- handler = self._filterModel.parentModel().handler()
- if (
- self._filterModel.parentModel().Type
- == InstancesModelTypes.User
- ):
- userInstances = True
- menu.addSeparator()
- delUserInstanceAction = menu.addAction(_("Remove selected"))
- updateUserInstanceAction = menu.addAction(_("Update selected"))
- if handler.hasActiveJobs():
- delUserInstanceAction.setEnabled(False)
- columnIndex = 0
- for columnData in self.model().getColumns():
- action = QWidgetAction(columnsMenu)
- widget = QCheckBox(columnData.name, columnsMenu)
- widget.setTristate(False)
- action.setDefaultWidget(widget)
-
- if not self.isColumnHidden(columnIndex):
- widget.setChecked(True)
- columnsMenu.addAction(action)
- widget.stateChanged.connect(
- lambda state, index=columnIndex:
- self._hideColumn(index, state)
- )
- columnIndex += 1
- action = menu.exec_(self.mapToGlobal(event.pos()))
- if action == addToBlacklistAction:
-
- indexes = self.selectionModel().selectedRows()
- forAll = None
- for index in indexes:
- url = self.model().getByIndex(index.row())
- if url in self._filterModel.whitelist:
-
- if forAll == QMessageBox.YesRole:
- __removeFromWhitelist(self, url)
- elif forAll == QMessageBox.NoRole:
- continue
- else:
- questionBox = QMessageBox(self)
- questionBox.addButton(
- _("Yes"), QMessageBox.YesRole
- )
- questionBox.addButton(
- _("No"), QMessageBox.NoRole
- )
- questionBox.addButton(
- _("Cancel"), QMessageBox.RejectRole
- )
- questionBox.setWindowTitle("Url whitelisted")
- questionBox.setText(_(
- f"{url} found in the <b>whitelist</b>. Would " \
- "you like to <b>remove</b> it from the" \
- " <b>whitelist</b> and add it to the" \
- " <b>blacklist</b>?")
- )
- checkBox = QCheckBox(
- _("Remember for all"), questionBox
- )
- questionBox.setCheckBox(checkBox)
- questionBox.exec()
- answer = questionBox.result()
- if answer == 0:
-
-
- __removeFromWhitelist(self, url)
- if checkBox.isChecked():
- forAll = QMessageBox.YesRole
- elif answer == 1 and checkBox.isChecked():
- forAll = QMessageBox.NoRole
- continue
- else:
- continue
- self._filterModel.putInstanceOnBlacklist(
- url, reason=_("Manual")
- )
- self._filterModel.apply()
- elif action == addToTmpBlacklistAction:
- index = selectionModel.selectedRows()[0]
- url = self.model().getByIndex(index.row())
- self._filterModel.putInstanceOnTimeout(url, reason=_("Manual"))
- self._filterModel.apply()
- elif action == addToWhitelistAction:
-
- indexes = self.selectionModel().selectedRows()
- for index in indexes:
- url = self.model().getByIndex(index.row())
- if url not in self._filterModel.whitelist:
- self._filterModel.putInstanceOnWhitelist(url)
- self._filterModel.apply()
- elif action == selectAllAction:
- self.selectAll()
-
- elif action in copyMenuActions:
- columnIndex = copyMenuActions.index(action)
- indexes = self.selectionModel().selectedRows()
- clipboard = QGuiApplication.clipboard()
- if len(indexes) > 1:
-
- result = []
- for index in indexes:
- rowIndex = index.row()
- newIndex = self.model().index(rowIndex, columnIndex)
- data = self.model().data(newIndex, Qt.DisplayRole)
- result.append(data)
- clipboard.setText(str(result))
- else:
-
- rowIndex = indexes[0].row()
- newIndex = self.model().index(rowIndex, columnIndex)
- data = self.model().data(newIndex, Qt.DisplayRole)
- clipboard.setText(str(data))
-
- elif action == copyJson:
- indexes = self.selectionModel().selectedRows()
- instancesModel = self.model()._model
- result = {}
- for index in indexes:
- url = self.model().data(index, Qt.DisplayRole)
- instance = instancesModel[url]
- result.update({instance.url: instance.data})
- clipboard = QGuiApplication.clipboard()
- clipboard.setText(json.dumps(result))
-
- if userInstances:
- if action == delUserInstanceAction:
- indexes = self.selectionModel().selectedRows()
-
- instancesStr = ""
- for index in indexes:
- url = self.model().getByIndex(index.row())
- instancesStr = f"{instancesStr}\n{url}"
- confirmDialog = QMessageBox()
- confirmDialog.setWindowTitle(
- _("Delete instances")
- )
- confirmDialog.setText(
- _("Are you sure you want to delete the following"
- " instances?\n") + instancesStr
- )
- confirmDialog.setStandardButtons(
- QMessageBox.Yes | QMessageBox.No
- )
- confirmDialog.button(QMessageBox.Yes).setText(_("Yes"))
- confirmDialog.button(QMessageBox.No).setText(_("No"))
- if confirmDialog.exec() != QMessageBox.Yes:
- return
- toRemove = []
- for index in indexes:
- toRemove.append(self.model().getByIndex(index.row()))
- handler.removeMultiInstances(toRemove)
- elif action == updateUserInstanceAction:
- indexes = self.selectionModel().selectedRows()
- handler = self._filterModel.parentModel().handler()
- for index in indexes:
- handler.updateInstance(
- self.model().getByIndex(index.row())
- )
- class UrlFilterLabel(QLabel):
- """ Fancy QLabel that eludes text and has a hover color property.
- """
- def __init__(self, text, parent):
- QLabel.__init__(self, text, parent=parent)
- self.__hover = False
- self.__hoverColor = QColor()
- self.setMinimumWidth(100)
- @pyqtProperty(QColor)
- def hoverColor(self):
- return self.__hoverColor
- @hoverColor.setter
- def hoverColor(self, color):
- self.__hoverColor = color
- def enterEvent(self, event):
- self.__hover = True
- self.update()
- def leaveEvent(self, event):
- self.__hover = False
- self.update()
- def paintEvent(self, event):
- painter = QPainter(self)
- if self.__hover and self.hoverColor.isValid():
- pen = painter.pen()
- pen.setColor(self.hoverColor)
- painter.setPen(pen)
- metrics = QFontMetrics(self.font())
- elided = metrics.elidedText(self.text(), Qt.ElideRight, self.width())
- painter.drawText(self.rect(), self.alignment(), elided)
- class UrlFilterItem(QWidget):
- deleted = pyqtSignal(str)
- def __init__(self, url, parent=None):
- QWidget.__init__(self, parent=parent)
- layout = QHBoxLayout(self)
- layout.setContentsMargins(0, 0, 0, 0)
- self._urlStr = url
- label = UrlFilterLabel(url, self)
- delButton = Button("X", self)
- delButton.setSizePolicy(
- QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed)
- )
- layout.addWidget(label, 0, Qt.AlignTop)
- layout.addWidget(delButton, 0, Qt.AlignTop)
- delButton.clicked.connect(self.__delMe)
- def str(self): return self._urlStr
- def __delMe(self):
- self.deleted.emit(self.str())
- self.deleteLater()
- class UrlFilterWidget(QWidget):
- def __init__(self, model, parent=None):
- QWidget.__init__(self, parent=parent)
- self._model = model
- self._items = {}
- QVBoxLayout(self)
- self.layout().setContentsMargins(0, 0, 0, 0)
- self.layout().setSpacing(0)
- self._model.changed.connect(self.__modelChanged)
-
- def filterItems(self):
- return []
- def delete(self, url):
- self._delete(url)
- def itemAdded(self, url):
- """ Callback from when a item has been added, usefull for subclasses
- to do something; for example set a tooltip.
- """
- pass
-
- def __modelChanged(self):
-
- for url in list(self._items.keys()):
- if url not in self.filterItems():
- self._items[url].deleteLater()
- del self._items[url]
-
- for url in self.filterItems():
- if url not in self._items:
- self._add(url)
- def _add(self, url):
- item = UrlFilterItem(url, self)
- self._items.update({url: item})
- item.deleted.connect(self.delete)
- self.layout().addWidget(item)
- self.itemAdded(url)
- def _delete(self, url):
- del self._items[url]
- class WhitelistFilterWidget(UrlFilterWidget):
- def filterItems(self):
- return self._model.whitelist
- def delete(self, url):
- UrlFilterWidget.delete(self, url)
- self._model.delInstanceFromWhitelist(url)
- self._model.apply()
- class BlacklistFilterWidget(UrlFilterWidget):
- def filterItems(self):
- return self._model.blacklist
- def itemAdded(self, url):
- filterItems = self.filterItems()
- dateStr = timeToString(filterItems[url][0] * 60)
- reasonStr = filterItems[url][1]
- tooltipStr = f"<b>{_('Date')}</b>: {dateStr}\n" \
- f"<b>{_('Reason')}</b>: {reasonStr}"
- self._items[url].setToolTip(tooltipStr)
- def delete(self, url):
- UrlFilterWidget.delete(self, url)
- self._model.delInstanceFromBlacklist(url)
- self._model.apply()
- class TimeoutFilterWidget(UrlFilterWidget):
- def filterItems(self):
- return self._model.timeoutList
- def itemAdded(self, url):
-
- durationStr = _("Until restart or manual removal.")
- filterItems = self.filterItems()
- timer = filterItems[url][0]
- if timer:
- durationStr = str(timer.interval() / 60000) + "min"
- reason = filterItems[url][1]
- tooltipStr = f"<b>{_('Reason')}</b>: \n{reason}\n<b>{_('Duration')}" \
- f"</b>: {durationStr}"
- self._items[url].setToolTip(tooltipStr)
- def delete(self, url):
- UrlFilterWidget.delete(self, url)
- self._model.delInstanceFromTimeout(url)
- self._model.apply()
- class NetworkFilterWidget(QWidget):
- def __init__(self, model, parent=None):
- """
- @type model: InstancesModel
- """
- QWidget.__init__(self, parent=parent)
- self._model = model
- layout = QHBoxLayout(self)
- layout.setContentsMargins(0, 0, 0, 0)
- layout.setSpacing(0)
- self._webBox = QCheckBox("Web", self)
- self._torBox = QCheckBox("Tor", self)
- self._i2pBox = QCheckBox("i2p", self)
- self._webBox.setSizePolicy(
- QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed)
- )
- self._torBox.setSizePolicy(
- QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed)
- )
- self._i2pBox.setSizePolicy(
- QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed)
- )
- layout.addWidget(self._webBox, 0, Qt.AlignLeft)
- layout.addWidget(self._torBox, 0, Qt.AlignLeft)
- layout.addWidget(self._i2pBox, 1, Qt.AlignLeft)
- self._model.changed.connect(self._modelChanged)
- self._webBox.stateChanged.connect(self.__changed)
- self._torBox.stateChanged.connect(self.__changed)
- self._i2pBox.stateChanged.connect(self.__changed)
- def __changed(self, state):
- self._model.changed.disconnect(self._modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self._modelChanged)
- def _modelChanged(self):
- self._webBox.stateChanged.disconnect(self.__changed)
- self._torBox.stateChanged.disconnect(self.__changed)
- self._i2pBox.stateChanged.disconnect(self.__changed)
- self._webBox.setChecked(False)
- self._torBox.setChecked(False)
- self._i2pBox.setChecked(False)
- filter = self._model.filter()
- if NetworkTypes.Web in filter['networkTypes']:
- self._webBox.setChecked(True)
- if NetworkTypes.Tor in filter['networkTypes']:
- self._torBox.setChecked(True)
- if NetworkTypes.I2P in filter['networkTypes']:
- self._i2pBox.setChecked(True)
- self._webBox.stateChanged.connect(self.__changed)
- self._torBox.stateChanged.connect(self.__changed)
- self._i2pBox.stateChanged.connect(self.__changed)
- def data(self):
- result = {
- 'networkTypes': []
- }
- if self._webBox.isChecked():
- result['networkTypes'].append(NetworkTypes.Web)
- if self._torBox.isChecked():
- result['networkTypes'].append(NetworkTypes.Tor)
- if self._i2pBox.isChecked():
- result['networkTypes'].append(NetworkTypes.I2P)
- return result
- class VersionFilter(QWidget):
- def __init__(self, model, parent=None):
- """
- @type model: InstanceModelFilter
- """
- QWidget.__init__(self, parent=parent)
- self._model = model
- self._items = {}
- layout = QVBoxLayout(self)
- layout.setContentsMargins(0, 0, 0, 0)
- layout.setSpacing(0)
- minVerLayout = QHBoxLayout()
- self.minVersionCombo = QComboBox(self)
- self.minVersionCombo.setSizePolicy(
- QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Fixed))
- minLabel = QLabel(_("Minimum:"), self)
- minLabel.setSizePolicy(
- QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Fixed))
- minVerLayout.addWidget(minLabel, 0, Qt.AlignLeft)
- minVerLayout.addWidget(self.minVersionCombo)
- self.invalidCheck = QCheckBox(_("Invalid"), self)
- self.invalidCheck.setToolTip(
- _("Include instances with a invalid version")
- )
- self.gitVersionCheck = QCheckBox(_("Development"), self)
- self.gitVersionCheck.setToolTip(
- _("Include development versions of SearX/SearXNG (git versions).")
- )
- self.dirtyVersionFlagCheck = QCheckBox(_("Dirty"), self)
- self.dirtyVersionFlagCheck.setToolTip(
- _("Include SearXNG git versions with uncommited changes.")
- )
-
- self.dirtyVersionFlagCheck.setEnabled(False)
- self.extraVersionFlagCheck = QCheckBox(_("Extra"), self)
- self.extraVersionFlagCheck.setToolTip(
- _("Include versions flagged as extra (IDK what that is? TODO).")
- )
-
- self.extraVersionFlagCheck.setEnabled(False)
- self.unknownVersionFlagCheck = QCheckBox(_("Unknown"), self)
- self.unknownVersionFlagCheck.setToolTip(
- _("Include Searx versions with unknown git commit.")
- )
-
- self.unknownVersionFlagCheck.setEnabled(False)
-
- layout.addLayout(minVerLayout, 1)
- layout.addWidget(self.invalidCheck)
- layout.addWidget(self.gitVersionCheck)
- layout.addWidget(self.dirtyVersionFlagCheck)
- layout.addWidget(self.extraVersionFlagCheck)
- layout.addWidget(self.unknownVersionFlagCheck)
-
- self._model.changed.connect(self.__modelChanged)
- self.minVersionCombo.currentTextChanged.connect(
- self._minVersionComboChanged
- )
- self.invalidCheck.stateChanged.connect(self._invalidChanged)
- self.gitVersionCheck.stateChanged.connect(self._gitChanged)
- self.dirtyVersionFlagCheck.stateChanged.connect(self._dirtyChanged)
- self.extraVersionFlagCheck.stateChanged.connect(self._extraChanged)
- self.unknownVersionFlagCheck.stateChanged.connect(self._unknownChanged)
- def _minVersionComboChanged(self, versionStr):
- """ From the ComboBox
- """
- self._model.changed.disconnect(self.__modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self.__modelChanged)
- def _invalidChanged(self, state):
- self._model.changed.disconnect(self.__modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self.__modelChanged)
- def _gitChanged(self, state):
- self._model.changed.disconnect(self.__modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self.__modelChanged)
- enabled = bool(state)
- self.dirtyVersionFlagCheck.setEnabled(enabled)
- self.extraVersionFlagCheck.setEnabled(enabled)
- self.unknownVersionFlagCheck.setEnabled(enabled)
- def _dirtyChanged(self, state):
- self._model.changed.disconnect(self.__modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self.__modelChanged)
- def _extraChanged(self, state):
- self._model.changed.disconnect(self.__modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self.__modelChanged)
- def _unknownChanged(self, state):
- self._model.changed.disconnect(self.__modelChanged)
- self._model.updateKwargs(self.data())
- self._model.changed.connect(self.__modelChanged)
- def _compileVersions(self):
- """ Fill the combobox with available version strings
- """
-
- versionFilter = self._model.filter().get('version', {})
- selectedText = versionFilter.get('min', "")
-
- self.minVersionCombo.clear()
-
- self.minVersionCombo.addItem(_("Off"))
-
- uniqueVersions = []
- for url, instance in self._model.parentModel().items():
- version = instance.version
- partsString = version.partsString()
- if version.isValid() and partsString not in uniqueVersions:
- uniqueVersions.append(partsString)
-
- uniqueVersions.sort()
-
- selectedIndex = 0
- for versionStr in uniqueVersions:
- if versionStr == selectedText:
-
-
- selectedIndex = self.minVersionCombo.count()
- self.minVersionCombo.addItem(versionStr)
-
- self.minVersionCombo.setCurrentIndex(selectedIndex)
- def __modelChanged(self):
-
-
- self.minVersionCombo.currentTextChanged.disconnect(
- self._minVersionComboChanged
- )
- self._compileVersions()
- self.minVersionCombo.currentTextChanged.connect(
- self._minVersionComboChanged
- )
- versionFilter = self._model.filter().get('version', {})
- self.gitVersionCheck.setChecked(versionFilter.get('git', True))
- self.dirtyVersionFlagCheck.setChecked(
- versionFilter.get('dirty', False)
- )
- self.extraVersionFlagCheck.setChecked(
- versionFilter.get('extra', False)
- )
- self.unknownVersionFlagCheck.setChecked(
- versionFilter.get('unknown', False)
- )
- self._compileVersions()
- def data(self):
- result = {
- 'version': {
- 'min': self.minVersionCombo.currentText(),
- 'git': bool(self.gitVersionCheck.checkState()),
- 'dirty': bool(self.dirtyVersionFlagCheck.checkState()),
- 'extra': bool(self.extraVersionFlagCheck.checkState()),
- 'unknown': bool(self.unknownVersionFlagCheck.checkState()),
- 'invalid': bool(self.invalidCheck.checkState())
- }
- }
- return result
- class CheckboxFilter(QCheckBox):
- def __init__(self, model, key, parent=None):
- """
- @type model: InstancesModel
- @param key: key to use (from model.filter())
- @type key: str
- """
- QCheckBox.__init__(self, parent=parent)
- self._model = model
- self._key = key
- self._model.changed.connect(self.__modelChanged)
- self.stateChanged.connect(self.__stateChanged)
- def __modelChanged(self):
- """ Model changed, update the checkbox.
- """
- filter_ = self._model.filter()
- state = filter_.get(self._key, True)
- self.stateChanged.disconnect(self.__stateChanged)
- self.setChecked(state)
- self.stateChanged.connect(self.__stateChanged)
- def __stateChanged(self):
- """ Checkbox state changed, update the model.
- """
- state = self.isChecked()
- self._model.updateKwargs({self._key: state})
- class FilterWidget(QWidget):
- def __init__(self, model, parent=None):
- """
- @type model: searxqt.models.instances.InstanceModelFilter
- """
- QWidget.__init__(self, parent=parent)
- self._model = model
- layout = QFormLayout(self)
- layout.setLabelAlignment(
- Qt.AlignLeading | Qt.AlignLeft | Qt.AlignTop
- )
- layout.addRow(QLabel(_("<h3>Filter</h3>"), self))
- self._networkFilter = NetworkFilterWidget(model, self)
- layout.addRow(_("Network") + ":", self._networkFilter)
- self._asnPrivacyFilter = None
- self._ipv6Filter = None
- self._analyticsFilter = None
- self._versionFilter = VersionFilter(model, self)
- layout.addRow(_("Version") + ":", self._versionFilter)
- self._urlFilterOut = BlacklistFilterWidget(model, self)
- layout.addRow(_("Blacklist") + ":", self._urlFilterOut)
- self._urlFilter = WhitelistFilterWidget(model, self)
- layout.addRow(_("Whitelist") + ":", self._urlFilter)
- self._tmpBlacklist = TimeoutFilterWidget(model, self)
- layout.addRow(_("Timeout") + ":", self._tmpBlacklist)
- model.parentModel().typeChanged.connect(self.__modelTypeChanged)
- def __modelTypeChanged(self, type_):
- layout = self.layout()
- if type_ == InstancesModelTypes.Stats2:
- if not self._asnPrivacyFilter:
- self._asnPrivacyFilter = CheckboxFilter(
- self._model,
- 'asnPrivacy',
- self
- )
- label = QLabel(_("Require ASN privacy") + ":")
- label.setWordWrap(True)
- label.setToolTip(_(
- "Filter out instances that run their server at a known\n"
- "malicious network like Google, CloudFlare, Akamai etc..\n"
- "\n"
- "This does not give any guarantee, it only filters known\n"
- "privacy violators!"
- ))
- layout.insertRow(2, label, self._asnPrivacyFilter)
- if not self._ipv6Filter:
- self._ipv6Filter = CheckboxFilter(self._model, 'ipv6', self)
- label = QLabel(_("Require IPv6") + ":")
- label.setWordWrap(True)
- label.setToolTip(_(
- "Only instances with at least one ipv6-address remain"
- " if checked."
- ))
- layout.insertRow(3, label, self._ipv6Filter)
- if not self._analyticsFilter:
- self._analyticsFilter = CheckboxFilter(
- self._model,
- 'analytics',
- self
- )
- label = QLabel(_("Excl. analytics") + ":")
- label.setWordWrap(True)
- label.setToolTip(_(
- "Exclude instances running tracking software."
- ))
- layout.insertRow(4, label, self._analyticsFilter)
- else:
- if self._asnPrivacyFilter:
- self._asnPrivacyFilter.deleteLater()
- layout.removeRow(2)
- self._asnPrivacyFilter = None
- if self._ipv6Filter:
- self._ipv6Filter.deleteLater()
- layout.removeRow(2)
- self._ipv6Filter = None
- if self._analyticsFilter:
- self._analyticsFilter.deleteLater()
- layout.removeRow(2)
- self._analyticsFilter = None
- class InstancesStatsWidget(QWidget):
- def __init__(self, filterModel, parent=None):
- """
- @param filterModel:
- @type filterModel: searxqt.models.instances.InstanceModelFilter
- """
- QWidget.__init__(self, parent=parent)
- self._filterModel = filterModel
- self._currentInstancesType = filterModel.parentModel().Type
- self._countLabel = None
- self._filterCountLabel = None
- self._lastUpdateLabel = None
- layout = QFormLayout(self)
- layout.setLabelAlignment(
- Qt.AlignLeading | Qt.AlignLeft | Qt.AlignTop
- )
- self._toggleButton = Button("", self)
- layout.addRow(_("<h3>Stats</h3>"), self._toggleButton)
- self.__show()
- self._toggleButton.clicked.connect(self.__toggle)
- self._filterModel.parentModel().typeChanged.connect(
- self.__instancesModelTypeChanged
- )
- def collapsed(self):
- return bool(self._countLabel is None)
- def __toggle(self):
- if self.collapsed():
- self.__show()
- else:
- self.__hide()
- def __show(self):
- self._toggleButton.setText(_("Hide"))
- layout = self.layout()
- self._countLabel = QLabel("0", self)
- layout.addRow(_("Total count") + ":", self._countLabel)
- self._filterCountLabel = QLabel("0", self)
- layout.addRow(_("After filter count") + ":", self._filterCountLabel)
- self._lastUpdateLabel = QLabel("0", self)
- self._lastUpdateLabel.setWordWrap(True)
- layout.addRow(_("Last update") + ":", self._lastUpdateLabel)
- self.__modelChanged()
- self.__instancesModelTypeChanged(self._filterModel.parentModel().Type)
- self._filterModel.changed.connect(self.__modelChanged)
- def __hide(self):
- self._filterModel.changed.disconnect(self.__modelChanged)
- self.__instancesModelTypeChanged(InstancesModelTypes.NotDefined)
- self._toggleButton.setText(_("Show"))
- layout = self.layout()
- while layout.rowCount() > 1:
- layout.removeRow(1)
- self._countLabel = None
- self._filterCountLabel = None
- self._lastUpdateLabel = None
- """ InstancesType dependant methods below
- """
- def __instancesModelTypeChanged(self, type_):
- """ PersistentInstancesModel emitted changed.
- """
- if self.collapsed():
- return
- previousType = self._currentInstancesType
- self._currentInstancesType = type_
-
- if previousType == InstancesModelTypes.Stats2:
- self._filterModel.changed.disconnect(self.__updateLastUpdated)
- elif previousType == InstancesModelTypes.User:
- pass
- if type_ == InstancesModelTypes.Stats2:
- self._filterModel.changed.connect(self.__updateLastUpdated)
- self.__updateLastUpdated()
- else:
- self._lastUpdateLabel.setText("-")
-
- def __updateLastUpdated(self):
- self._lastUpdateLabel.setText(
- timeToString(
- self._filterModel.parentModel().handler().lastUpdated()
- )
- )
- """ -------------------------------------------
- """
- def __modelChanged(self):
- self._countLabel.setText(str(len(self._filterModel.parentModel())))
- self._filterCountLabel.setText(str(len(self._filterModel)))
|