RepoMirrors/setools
1
0
Fork 0
mirror of https://github.com/SELinuxProject/setools synced 2025-04-11 03:51:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

ApolMainWindow: Do not save tab/workspace if there are errors.

Browse Source 
Tabs should raise RuntimeError in save() if they have criteria errors.

For #97 and #98.
This commit is contained in:
Chris PeBenito 2016-07-01 22:14:35 -04:00
parent a82be1fbdb
commit 09bca896da
1 changed files with 34 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
setoolsgui/apol/mainwindow.py
View File

@ -410,6 +410,16 @@ class ApolMainWindow(SEToolsWidget, QMainWindow):
self.load_settings(new=True)
def save_settings(self):
try:
settings = self._get_settings()
except RuntimeError:
self.log.critical("Errors in the query prevent saving the settings.")
self.error_msg.critical(self, "Unable to save settings",
"Please resolve errors in the query before saving the settings."
)
return
filename = QFileDialog.getSaveFileName(self, "Save analysis tab settings", "analysis.apolt",
"Apol Tab Settings File (*.apolt);;"
"All Files (*)")[0]
@ -417,8 +427,6 @@ class ApolMainWindow(SEToolsWidget, QMainWindow):
if not filename:
return
settings = self._get_settings()
try:
with open(filename, "w") as fd:
json.dump(settings, fd, indent=1)
@ -556,6 +564,30 @@ class ApolMainWindow(SEToolsWidget, QMainWindow):
format("\n\n".join(loading_errors)))
def save_workspace(self):
workspace = {}
try:
workspace["__policy__"] = os.path.abspath(str(self._policy))
workspace["__permmap__"] = os.path.abspath(str(self._permmap))
workspace["__tabs__"] = []
for index in range(self.AnalysisTabs.count()):
tab = self.AnalysisTabs.widget(index)
settings = tab.save()
# add the tab info to the settings.
settings["__title__"] = self.AnalysisTabs.tabText(index)
settings["__tab__"] = type(tab).__name__
workspace["__tabs__"].append(settings)
except RuntimeError:
self.log.critical("Errors in the query prevent saving the workspace.")
self.error_msg.critical(self, "Unable to save workspace",
"Please resolve errors in tab \"{0}\" before saving the"
" settings.".format(self.AnalysisTabs.tabText(index)))
return
filename = QFileDialog.getSaveFileName(self, "Save analysis workspace", "workspace.apolw",
"Apol Workspace Files (*.apolw);;"
"All Files (*)")[0]
@ -563,22 +595,6 @@ class ApolMainWindow(SEToolsWidget, QMainWindow):
if not filename:
return
workspace = {}
workspace["__policy__"] = os.path.abspath(str(self._policy))
workspace["__permmap__"] = os.path.abspath(str(self._permmap))
workspace["__tabs__"] = []
for index in range(self.AnalysisTabs.count()):
tab = self.AnalysisTabs.widget(index)
settings = tab.save()
# add the tab info to the settings.
settings["__title__"] = self.AnalysisTabs.tabText(index)
settings["__tab__"] = type(tab).__name__
workspace["__tabs__"].append(settings)
with open(filename, "w") as fd:
json.dump(workspace, fd, indent=1)