mirror of
https://github.com/SELinuxProject/setools
synced 2025-04-23 23:55:18 +00:00
b71f9eda92
Prevent crashes in the unlikely event there is an exception while a query is processing. Signed-off-by: Chris PeBenito <pebenito@ieee.org>
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
# Copyright 2016, Tresys Technology, LLC
|
|
#
|
|
# SPDX-License-Identifier: LGPL-2.1-only
|
|
#
|
|
#
|
|
import logging
|
|
from typing import TYPE_CHECKING
|
|
|
|
from PyQt5 import QtCore
|
|
|
|
if TYPE_CHECKING:
|
|
from setools.query import PolicyQuery
|
|
from .models.table import SEToolsTableModel
|
|
|
|
|
|
class QueryResultsUpdater(QtCore.QObject):
|
|
|
|
"""
|
|
Thread for processing basic queries and updating result widgets.
|
|
|
|
Parameters:
|
|
query The query object
|
|
model The model for the results
|
|
|
|
Qt signals:
|
|
failed (str) The updated failed, with an error message.
|
|
finished (int) The update has completed, with the number of results.
|
|
raw_line (str) A string to be appended to the raw results.
|
|
"""
|
|
|
|
failed = QtCore.pyqtSignal(str)
|
|
finished = QtCore.pyqtSignal(int)
|
|
raw_line = QtCore.pyqtSignal(str)
|
|
|
|
def __init__(self, query: "PolicyQuery", model: "SEToolsTableModel") -> None:
|
|
super().__init__()
|
|
self.query = query
|
|
self.model = model
|
|
self.log = logging.getLogger(self.query.__module__)
|
|
|
|
def update(self) -> None:
|
|
"""Run the query and update results."""
|
|
results = []
|
|
counter = 0
|
|
|
|
try:
|
|
for counter, item in enumerate(self.query.results(), start=1):
|
|
results.append(item)
|
|
|
|
self.raw_line.emit(str(item))
|
|
|
|
if QtCore.QThread.currentThread().isInterruptionRequested():
|
|
break
|
|
elif counter % 10 == 0:
|
|
# yield execution every 10 rules
|
|
QtCore.QThread.yieldCurrentThread()
|
|
|
|
if counter % 1000 == 0:
|
|
self.log.info(f"Generated {counter} results so far.")
|
|
|
|
self.log.info(f"Generated {counter} total results.")
|
|
|
|
except Exception as e:
|
|
self.failed.emit(str(e))
|
|
|
|
else:
|
|
self.model.item_list = results
|
|
self.finished.emit(counter)
|