mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-06-01 18:49:38 +12:00
Fix crash when regex is not valid (Fixes olivierkes#989)
Wrap regex preparation in try-catch Log when there is a problem preparing the regex
This commit is contained in:
parent
0182a43b8e
commit
795e1fa9c0
|
@ -13,6 +13,8 @@ from manuskript.enums import Model
|
||||||
from manuskript.models.flatDataModelWrapper import flatDataModelWrapper
|
from manuskript.models.flatDataModelWrapper import flatDataModelWrapper
|
||||||
from manuskript.ui.searchMenu import searchMenu
|
from manuskript.ui.searchMenu import searchMenu
|
||||||
from manuskript.ui.highlighters.searchResultHighlighters.searchResultHighlighter import searchResultHighlighter
|
from manuskript.ui.highlighters.searchResultHighlighters.searchResultHighlighter import searchResultHighlighter
|
||||||
|
import logging
|
||||||
|
LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class search(QWidget, Ui_search):
|
class search(QWidget, Ui_search):
|
||||||
|
@ -69,21 +71,30 @@ class search(QWidget, Ui_search):
|
||||||
self.openItem(self.result.currentItem())
|
self.openItem(self.result.currentItem())
|
||||||
|
|
||||||
def prepareRegex(self, searchText):
|
def prepareRegex(self, searchText):
|
||||||
import re
|
rtn = None
|
||||||
|
try:
|
||||||
|
import re
|
||||||
|
|
||||||
flags = re.UNICODE
|
flags = re.UNICODE
|
||||||
|
|
||||||
if self.searchMenu.caseSensitive() is False:
|
if self.searchMenu.caseSensitive() is False:
|
||||||
flags |= re.IGNORECASE
|
flags |= re.IGNORECASE
|
||||||
|
|
||||||
if self.searchMenu.regex() is False:
|
if self.searchMenu.regex() is False:
|
||||||
searchText = re.escape(searchText)
|
searchText = re.escape(searchText)
|
||||||
|
|
||||||
if self.searchMenu.matchWords() is True:
|
if self.searchMenu.matchWords() is True:
|
||||||
# Source: https://stackoverflow.com/a/15863102
|
# Source: https://stackoverflow.com/a/15863102
|
||||||
searchText = r'\b' + searchText + r'\b'
|
searchText = r'\b' + searchText + r'\b'
|
||||||
|
|
||||||
return re.compile(searchText, flags)
|
rtn = re.compile(searchText, flags)
|
||||||
|
except re.error as e:
|
||||||
|
LOGGER.info("Problem preparing regular expression: " + e.msg)
|
||||||
|
rtn = None
|
||||||
|
except Exception as e:
|
||||||
|
LOGGER.info("Problem preparing regular expression")
|
||||||
|
rtn = None
|
||||||
|
return rtn
|
||||||
|
|
||||||
def search(self):
|
def search(self):
|
||||||
self.result.clear()
|
self.result.clear()
|
||||||
|
@ -91,30 +102,33 @@ class search(QWidget, Ui_search):
|
||||||
|
|
||||||
searchText = self.searchTextInput.text()
|
searchText = self.searchTextInput.text()
|
||||||
if len(searchText) > 0:
|
if len(searchText) > 0:
|
||||||
|
results = list()
|
||||||
searchRegex = self.prepareRegex(searchText)
|
searchRegex = self.prepareRegex(searchText)
|
||||||
results = []
|
if searchRegex is not None:
|
||||||
|
# Set override cursor
|
||||||
|
qApp.setOverrideCursor(Qt.WaitCursor)
|
||||||
|
|
||||||
# Set override cursor
|
for model, modelName in [
|
||||||
qApp.setOverrideCursor(Qt.WaitCursor)
|
(mainWindow().mdlOutline, Model.Outline),
|
||||||
|
(mainWindow().mdlCharacter, Model.Character),
|
||||||
|
(flatDataModelWrapper(mainWindow().mdlFlatData), Model.FlatData),
|
||||||
|
(mainWindow().mdlWorld, Model.World),
|
||||||
|
(mainWindow().mdlPlots, Model.Plot)
|
||||||
|
]:
|
||||||
|
filteredColumns = self.searchMenu.columns(modelName)
|
||||||
|
|
||||||
for model, modelName in [
|
# Searching
|
||||||
(mainWindow().mdlOutline, Model.Outline),
|
if len(filteredColumns):
|
||||||
(mainWindow().mdlCharacter, Model.Character),
|
results += model.searchOccurrences(searchRegex, filteredColumns)
|
||||||
(flatDataModelWrapper(mainWindow().mdlFlatData), Model.FlatData),
|
|
||||||
(mainWindow().mdlWorld, Model.World),
|
|
||||||
(mainWindow().mdlPlots, Model.Plot)
|
|
||||||
]:
|
|
||||||
filteredColumns = self.searchMenu.columns(modelName)
|
|
||||||
|
|
||||||
# Searching
|
# Showing results
|
||||||
if len(filteredColumns):
|
self.generateResultsLists(results)
|
||||||
results += model.searchOccurrences(searchRegex, filteredColumns)
|
|
||||||
|
|
||||||
# Showing results
|
# Remove override cursor
|
||||||
self.generateResultsLists(results)
|
qApp.restoreOverrideCursor()
|
||||||
|
else:
|
||||||
# Remove override cursor
|
# No results to generate if there is a problem with the regex
|
||||||
qApp.restoreOverrideCursor()
|
self.generateResultsLists(list())
|
||||||
|
|
||||||
def generateResultsLists(self, results):
|
def generateResultsLists(self, results):
|
||||||
self.noResultsLabel.setVisible(len(results) == 0)
|
self.noResultsLabel.setVisible(len(results) == 0)
|
||||||
|
@ -131,6 +145,7 @@ class search(QWidget, Ui_search):
|
||||||
def leaveEvent(self, event):
|
def leaveEvent(self, event):
|
||||||
self.delegate.mouseLeave()
|
self.delegate.mouseLeave()
|
||||||
|
|
||||||
|
|
||||||
class listResultDelegate(QStyledItemDelegate):
|
class listResultDelegate(QStyledItemDelegate):
|
||||||
def __init__(self, parent=None):
|
def __init__(self, parent=None):
|
||||||
QStyledItemDelegate.__init__(self, parent)
|
QStyledItemDelegate.__init__(self, parent)
|
||||||
|
|
Loading…
Reference in a new issue