def __init__(self, model_controller, plugin_manager, workspace_manager):
FaradayUi.__init__(self, model_controller, plugin_manager,
workspace_manager)
qt.QApplication.__init__(self, [])
self._shell_envs = dict()
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(), self,
self.getModelController(),
self.getPluginManager())
self.setMainWidget(self.getMainWindow())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.notification_center.registerWidget(self._main_window)
self.loghandler = GUIHandler()
addHandler(self.loghandler)
self._splash_screen = qt.QSplashScreen(
qt.QPixmap(os.path.join(CONF.getImagePath(), "splash2.png")),
qt.Qt.WStyle_StaysOnTop)
self.startSplashScreen()
def __init__(self, model_controller, plugin_manager, workspace_manager):
FaradayUi.__init__(self,
model_controller,
plugin_manager,
workspace_manager)
qt.QApplication.__init__(self, [])
self._shell_envs = dict()
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(),
self,
self.getModelController(),
self.getPluginManager())
self.setMainWidget(self.getMainWindow())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.notification_center.registerWidget(self._main_window)
self.loghandler = GUIHandler()
addHandler(self.loghandler)
self._splash_screen = qt.QSplashScreen(
qt.QPixmap(os.path.join(CONF.getImagePath(), "splash2.png")),
qt.Qt.WStyle_StaysOnTop)
def __init__(self):
self._original_excepthook = sys.excepthook
self.qapp = qt.QApplication([])
self._configuration = CONF
self._shell_envs = dict()
self._security_manager = SecurityManager()
self._model_controller = model.controller.ModelController(security_manager = self._security_manager)
self.plugin_manager = PluginManager(os.path.join(CONF.getConfigPath(),"plugins"))
self._workspace_manager = WorkspaceManager(self._model_controller,
self.plugin_manager.createController("ReportManager"))
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(), self, self._model_controller)
self.qapp.setMainWidget(self._main_window)
self._splash_screen = qt.QSplashScreen(qt.QPixmap(os.path.join(CONF.getImagePath(),"splash2.png")),
qt.Qt.WStyle_StaysOnTop)
if not self.getLogger().isGUIOutputRegistered():
self.logger.registerGUIOutput(self._main_window.getLogConsole())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.setMainApp(self)
class MainApplication(object):
"""
Contains the main QApplication to start the main loop
The application will handle one workspace at a time.
The workspace contains the list of hosts discovered
ShellEnvironments correspond to shell tabs
ModelController is the bridge between components and discovered
hosts & services
This class is the principal component, it is responsible for handling
some events coming from gui and connecting all components together
"""
logger = None
@staticmethod
def getLogger():
if MainApplication.logger is None:
MainApplication.logger = model.log.getLogger()
return MainApplication.logger
def __init__(self):
self._original_excepthook = sys.excepthook
self.qapp = qt.QApplication([])
self._configuration = CONF
self._shell_envs = dict()
self._security_manager = SecurityManager()
self._model_controller = model.controller.ModelController(security_manager = self._security_manager)
self.plugin_manager = PluginManager(os.path.join(CONF.getConfigPath(),"plugins"))
self._workspace_manager = WorkspaceManager(self._model_controller,
self.plugin_manager.createController("ReportManager"))
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(), self, self._model_controller)
self.qapp.setMainWidget(self._main_window)
self._splash_screen = qt.QSplashScreen(qt.QPixmap(os.path.join(CONF.getImagePath(),"splash2.png")),
qt.Qt.WStyle_StaysOnTop)
if not self.getLogger().isGUIOutputRegistered():
self.logger.registerGUIOutput(self._main_window.getLogConsole())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.setMainApp(self)
def enableExceptHook(self):
sys.excepthook = exception_handler
installThreadExcepthook()
def disableLogin(self):
CONF.setAuth(sys.disablelogin)
def start(self):
try:
splash_timer = qt.QTimer.singleShot(1700, lambda *args:None)
self._splash_screen.show()
signal.signal(signal.SIGINT, signal.SIG_DFL)
self._writeSplashMessage("Setting up remote API's...")
model.api.setUpAPIs(self._model_controller,CONF.getApiConInfoHost(),CONF.getApiConInfoPort())
model.guiapi.setUpGUIAPIs(self._model_controller)
self._writeSplashMessage("Starting model controller daemon...")
self._model_controller.start()
model.api.startAPIServer()
self._writeSplashMessage("Setting up main GUI...")
self._writeSplashMessage("Creating default shell...")
self._main_window.createShellTab()
self._writeSplashMessage("Ready...")
self.logger.log("Faraday ready...")
self._splash_screen.finish(self._main_window)
self._main_window.showAll()
logged = True
while True:
username, password = "******","password"
if username is None and password is None:
break
result = self._security_manager.authenticateUser(username, password)
if result == codes.successfulLogin:
logged = True
break
if logged:
self._main_window.showLoggedUser(self._security_manager.current_user.display_name)
model.api.__current_logged_user = username
self._workspace_manager.loadWorkspaces()
last_workspace = CONF.getLastWorkspace()
w = self._workspace_manager.createWorkspace(last_workspace)
self._workspace_manager.setActiveWorkspace(w)
self._main_window.getWorkspaceTreeView().loadAllWorkspaces()
self._workspace_manager.startReportManager()
except Exception:
print "There was an error while starting Faraday"
print "-" * 50
traceback.print_exc()
print "-" * 50
self.__exit(-1)
if logged:
exit_code = self.qapp.exec_loop()
else:
exit_code = -1
return self.__exit(exit_code)
def __exit(self, exit_code=0):
"""
Exits the application with the provided code.
It also waits until all app threads end.
"""
self._workspace_manager.stopAutoLoader()
self._workspace_manager.stopReportManager()
self._main_window.hide()
print "Closing Faraday..."
self._workspace_manager.saveWorkspaces()
envs = [env for env in self._shell_envs.itervalues()]
for env in envs:
env.terminate()
print "stopping model controller thread..."
self._model_controller.stop()
print "stopping model controller thread..."
self.qapp.quit()
print "Waiting for controller threads to end..."
self._model_controller.join()
return exit_code
def quit(self):
"""
Redefined quit handler to nicely end up things
"""
self.qapp.quit()
def createShellEnvironment(self, name = None):
model.api.devlog("createShellEnvironment called - About to create new shell env with name %s" % name)
shell_env = ShellEnvironment(name, self.qapp,
self._main_window.getTabManager(),
self._model_controller,
self.plugin_manager.createController,
self.deleteShellEnvironment)
self._shell_envs[name] = shell_env
self._main_window.addShell(shell_env.widget)
shell_env.run()
def deleteShellEnvironment(self, name, ref=None):
def _closeShellEnv(name):
try:
env = self._shell_envs[name]
env.terminate()
tabmanager.removeView(env.widget)
del self._shell_envs[name]
except Exception:
model.api.devlog("ShellEnvironment could not be deleted")
model.api.devlog("%s" % traceback.format_exc())
model.api.devlog("deleteShellEnvironment called - name = %s - ref = %r" % (name, ref))
tabmanager = self._main_window.getTabManager()
if len(self._shell_envs) > 1 :
_closeShellEnv(name)
else:
if ref is not None:
result = self._main_window.exitFaraday()
if result == qt.QDialog.Accepted:
self.quit()
else:
_closeShellEnv(name)
self._main_window.createShellTab()
def getMainWindow(self):
return self._main_window
def getWorkspaceManager(self):
return self._workspace_manager
def removeWorkspace(self, name):
model.api.log("Removing Workspace: %s" % name)
return self.getWorkspaceManager().removeWorkspace(name)
def syncWorkspaces(self):
try:
self._workspace_manager.saveWorkspaces()
except Exception:
model.api.log("An exception was captured while synchronizing workspaces\n%s"
% traceback.format_exc(), "ERROR")
def saveWorkspaces(self):
try:
self._workspace_manager.saveWorkspaces()
except Exception:
model.api.log("An exception was captured while saving workspaces\n%s"
% traceback.format_exc(), "ERROR")
def createWorkspace(self, name, description="", w_type=""):
if name in self._workspace_manager.getWorkspacesNames():
model.api.log("A workspace with name %s already exists"
% name, "ERROR")
else:
model.api.log("Creating workspace '%s'" % name)
model.api.devlog("Looking for the delegation class")
workingClass = globals()[w_type]
w = self._workspace_manager.createWorkspace(name, description, workspaceClass = workingClass )
self._workspace_manager.setActiveWorkspace(w)
self._model_controller.setWorkspace(w)
self._main_window.refreshWorkspaceTreeView()
self._main_window.getWorkspaceTreeView().loadAllWorkspaces()
def openWorkspace(self, name):
self.saveWorkspaces()
try:
workspace = self._workspace_manager.openWorkspace(name)
self._model_controller.setWorkspace(workspace)
except Exception:
model.api.log("An exception was captured while opening workspace %s\n%s"
% (name, traceback.format_exc()), "ERROR")
def _writeSplashMessage(self, text):
self._splash_screen.message(text, qt.Qt.AlignRight | qt.Qt.AlignTop, qt.Qt.red)
class GuiApp(qt.QApplication, FaradayUi):
def __init__(self, model_controller, plugin_manager, workspace_manager):
FaradayUi.__init__(self, model_controller, plugin_manager,
workspace_manager)
qt.QApplication.__init__(self, [])
self._shell_envs = dict()
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(), self,
self.getModelController(),
self.getPluginManager())
self.setMainWidget(self.getMainWindow())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.notification_center.registerWidget(self._main_window)
self._splash_screen = qt.QSplashScreen(
qt.QPixmap(os.path.join(CONF.getImagePath(), "splash2.png")),
qt.Qt.WStyle_StaysOnTop)
def getMainWindow(self):
return self._main_window
def run(self, args):
self._main_window.createShellTab()
self.createLoggerWidget()
self._main_window.showAll()
couchURL = CONF.getCouchURI()
if couchURL:
url = "%s/reports/_design/reports/index.html" % couchURL
model.api.log("Faraday ui is ready")
model.api.log(
"Make sure you have couchdb up and running if you want visualizations."
)
model.api.log("If couchdb is up, point your browser to: [%s]" %
url)
else:
model.api.log("Please configure Couchdb for fancy HTML5 Dashboard")
exit_code = self.exec_loop()
return exit_code
def createLoggerWidget(self):
if not model.log.getLogger().isGUIOutputRegistered():
model.log.getLogger().registerGUIOutput(
self._main_window.getLogConsole())
def loadWorkspaces(self):
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
splash_timer = qt.QTimer.singleShot(1700, lambda *args: None)
self._splash_screen.show()
def splashMessage(self, message):
self._splash_screen.message(message,
qt.Qt.AlignLeft | qt.Qt.AlignBottom,
qt.QColor(180, 0, 0))
def stopSplashScreen(self):
self._splash_screen.finish(self._main_window)
def quit(self):
model.log.getLogger().clearWidgets()
self.getMainWindow().hide()
envs = [env for env in self._shell_envs.itervalues()]
for env in envs:
env.terminate()
# exit status
notifier = model.log.getNotifier()
notifier.widget = None
qt.QApplication.quit(self)
def postEvent(self, receiver, event):
if receiver is None:
receiver = self.getMainWindow()
qt.QApplication.postEvent(receiver, QtCustomEvent.create(event))
def createShellEnvironment(self, name=None):
model.api.devlog("createShellEnvironment called \
- About to create new shell env with name %s" % name)
shell_env = ShellEnvironment(name, self,
self.getMainWindow().getTabManager(),
self.model_controller,
self.plugin_manager.createController,
self.deleteShellEnvironment)
self._shell_envs[name] = shell_env
self.getMainWindow().addShell(shell_env.widget)
shell_env.run()
def deleteShellEnvironment(self, name, ref=None):
def _closeShellEnv(name):
try:
env = self._shell_envs[name]
env.terminate()
tabmanager.removeView(env.widget)
del self._shell_envs[name]
except Exception:
model.api.devlog("ShellEnvironment could not be deleted")
model.api.devlog("%s" % traceback.format_exc())
model.api.devlog("deleteShellEnvironment called \
- name = %s - ref = %r" % (name, ref))
tabmanager = self.getMainWindow().getTabManager()
if len(self._shell_envs) > 1:
_closeShellEnv(name)
else:
if ref is not None:
result = self.getMainWindow().exitFaraday()
if result == qt.QDialog.Accepted:
self.quit()
else:
_closeShellEnv(name)
self.getMainWindow().createShellTab()
def removeWorkspace(self, name):
model.api.log("Removing Workspace: %s" % name)
return self.getWorkspaceManager().removeWorkspace(name)
def createWorkspace(self, name, description="", w_type=""):
if name in self.getWorkspaceManager().getWorkspacesNames():
model.api.log("A workspace with name %s already exists" % name,
"ERROR")
else:
model.api.log("Creating workspace '%s'" % name)
model.api.devlog("Looking for the delegation class")
manager = self.getWorkspaceManager()
w = manager.createWorkspace(name, description,
manager.namedTypeToDbType(w_type))
self.getMainWindow().refreshWorkspaceTreeView()
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def openWorkspace(self, name):
try:
self.getWorkspaceManager().openWorkspace(name)
except Exception:
model.api.log(
"An exception was captured while opening \
workspace %s\n%s" % (name, traceback.format_exc()), "ERROR")
class GuiApp(qt.QApplication, FaradayUi):
def __init__(self, model_controller, plugin_manager, workspace_manager):
FaradayUi.__init__(self,
model_controller,
plugin_manager,
workspace_manager)
qt.QApplication.__init__(self, [])
self._shell_envs = dict()
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(),
self,
self.getModelController(),
self.getPluginManager())
self.setMainWidget(self.getMainWindow())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.notification_center.registerWidget(self._main_window)
self.loghandler = GUIHandler()
addHandler(self.loghandler)
self._splash_screen = qt.QSplashScreen(
qt.QPixmap(os.path.join(CONF.getImagePath(), "splash2.png")),
qt.Qt.WStyle_StaysOnTop)
def getMainWindow(self):
return self._main_window
def run(self, args):
self.createLoggerWidget()
self._main_window.showAll()
couchURL = CONF.getCouchURI()
if couchURL:
url = "%s/reports/_design/reports/index.html" % couchURL
model.api.log("Faraday ui is ready")
model.api.log("Make sure you have couchdb up and running if you want visualizations.")
model.api.log("If couchdb is up, point your browser to: [%s]" % url)
else:
model.api.log("Please configure Couchdb for fancy HTML5 Dashboard (https://github.com/infobyte/faraday/wiki/Couchdb)")
exit_code = self.exec_loop()
return exit_code
def createLoggerWidget(self):
self.loghandler.registerGUIOutput(self._main_window.getLogConsole())
def loadWorkspaces(self):
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
splash_timer = qt.QTimer.singleShot(1700, lambda *args: None)
self._splash_screen.show()
def splashMessage(self, message):
self._splash_screen.message(
message,
qt.Qt.AlignLeft | qt.Qt.AlignBottom,
qt.QColor(180, 0, 0))
def stopSplashScreen(self):
self._splash_screen.finish(self._main_window)
def quit(self):
self.loghandler.clearWidgets()
self.getMainWindow().hide()
envs = [env for env in self._shell_envs.itervalues()]
for env in envs:
env.terminate()
# exit status
notifier = model.log.getNotifier()
notifier.widget = None
qt.QApplication.quit(self)
def postEvent(self, receiver, event):
if receiver is None:
receiver = self.getMainWindow()
qt.QApplication.postEvent(receiver, QtCustomEvent.create(event))
def createShellEnvironment(self, name=None):
model.api.devlog("createShellEnvironment called \
- About to create new shell env with name %s" % name)
shell_env = ShellEnvironment(name, self,
self.getMainWindow().getTabManager(),
self.model_controller,
self.plugin_manager.createController,
self.deleteShellEnvironment)
self._shell_envs[name] = shell_env
self.getMainWindow().addShell(shell_env.widget)
shell_env.run()
def deleteShellEnvironment(self, name, ref=None):
def _closeShellEnv(name):
try:
env = self._shell_envs[name]
env.terminate()
tabmanager.removeView(env.widget)
del self._shell_envs[name]
except Exception:
model.api.devlog("ShellEnvironment could not be deleted")
model.api.devlog("%s" % traceback.format_exc())
model.api.devlog("deleteShellEnvironment called \
- name = %s - ref = %r" % (name, ref))
tabmanager = self.getMainWindow().getTabManager()
if len(self._shell_envs) > 1:
_closeShellEnv(name)
else:
if ref is not None:
result = self.getMainWindow().exitFaraday()
if result == qt.QDialog.Accepted:
self.quit()
else:
_closeShellEnv(name)
self.getMainWindow().createShellTab()
def removeWorkspace(self, name):
model.api.log("Removing Workspace: %s" % name)
return self.getWorkspaceManager().removeWorkspace(name)
def createWorkspace(self, name, description="", w_type=""):
if name in self.getWorkspaceManager().getWorkspacesNames():
model.api.log("A workspace with name %s already exists"
% name, "ERROR")
else:
model.api.log("Creating workspace '%s'" % name)
model.api.devlog("Looking for the delegation class")
manager = self.getWorkspaceManager()
w = manager.createWorkspace(name, description,
manager.namedTypeToDbType(w_type))
self.getMainWindow().refreshWorkspaceTreeView()
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def openWorkspace(self, name):
try:
self.getWorkspaceManager().openWorkspace(name)
except Exception:
model.api.log("An exception was captured while opening \
workspace %s\n%s" % (name, traceback.format_exc()), "ERROR")
class GuiApp(qt.QApplication, FaradayUi):
def __init__(self, model_controller, plugin_manager, workspace_manager):
FaradayUi.__init__(self, model_controller, plugin_manager,
workspace_manager)
qt.QApplication.__init__(self, [])
self._shell_envs = dict()
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(), self,
self.getModelController(),
self.getPluginManager())
self.setMainWidget(self.getMainWindow())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.notification_center.registerWidget(self._main_window)
self._splash_screen = qt.QSplashScreen(
qt.QPixmap(os.path.join(CONF.getImagePath(), "splash2.png")),
qt.Qt.WStyle_StaysOnTop)
def getMainWindow(self):
return self._main_window
def run(self, args):
self._main_window.createShellTab()
self.createLoggerWidget()
self._main_window.showAll()
exit_code = self.exec_loop()
return exit_code
def createLoggerWidget(self):
if not model.log.getLogger().isGUIOutputRegistered():
model.log.getLogger().registerGUIOutput(
self._main_window.getLogConsole())
def loadWorkspaces(self):
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
splash_timer = qt.QTimer.singleShot(1700, lambda *args: None)
self._splash_screen.show()
def stopSplashScreen(self):
self._splash_screen.finish(self._main_window)
def quit(self):
model.log.getLogger().clearWidgets()
self.getMainWindow().hide()
envs = [env for env in self._shell_envs.itervalues()]
for env in envs:
env.terminate()
# exit status
notifier = model.log.getNotifier()
notifier.widget = None
qt.QApplication.quit(self)
def postEvent(self, receiver, event):
if receiver is None:
receiver = self.getMainWindow()
qt.QApplication.postEvent(receiver, QtCustomEvent.create(event))
def createShellEnvironment(self, name=None):
model.api.devlog("createShellEnvironment called \
- About to create new shell env with name %s" % name)
shell_env = ShellEnvironment(name, self,
self.getMainWindow().getTabManager(),
self.model_controller,
self.plugin_manager.createController,
self.deleteShellEnvironment)
self._shell_envs[name] = shell_env
self.getMainWindow().addShell(shell_env.widget)
shell_env.run()
def deleteShellEnvironment(self, name, ref=None):
def _closeShellEnv(name):
try:
env = self._shell_envs[name]
env.terminate()
tabmanager.removeView(env.widget)
del self._shell_envs[name]
except Exception:
model.api.devlog("ShellEnvironment could not be deleted")
model.api.devlog("%s" % traceback.format_exc())
model.api.devlog("deleteShellEnvironment called \
- name = %s - ref = %r" % (name, ref))
tabmanager = self.getMainWindow().getTabManager()
if len(self._shell_envs) > 1:
_closeShellEnv(name)
else:
if ref is not None:
result = self.getMainWindow().exitFaraday()
if result == qt.QDialog.Accepted:
self.quit()
else:
_closeShellEnv(name)
self.getMainWindow().createShellTab()
def removeWorkspace(self, name):
model.api.log("Removing Workspace: %s" % name)
return self.getWorkspaceManager().removeWorkspace(name)
def syncWorkspaces(self):
try:
self.getWorkspaceManager().saveWorkspaces()
except Exception:
model.api.log(
"An exception was captured while synchronizing \
workspaces\n%s" % traceback.format_exc(), "ERROR")
def saveWorkspaces(self):
try:
self.getWorkspaceManager().saveWorkspaces()
except Exception:
model.api.log(
"An exception was captured while saving \
workspaces\n%s" % traceback.format_exc(), "ERROR")
def createWorkspace(self, name, description="", w_type=""):
if name in self.getWorkspaceManager().getWorkspacesNames():
model.api.log("A workspace with name %s already exists" % name,
"ERROR")
else:
model.api.log("Creating workspace '%s'" % name)
model.api.devlog("Looking for the delegation class")
manager = self.getWorkspaceManager()
workingClass = None
if w_type and w_type in globals():
# If set as argument, otherwise let creation delegate behaviour
workingClass = globals()[w_type]
w = manager.createWorkspace(name,
description,
workspaceClass=workingClass)
self.getWorkspaceManager().setActiveWorkspace(w)
self.getModelController().setWorkspace(w)
self.getMainWindow().refreshWorkspaceTreeView()
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def openWorkspace(self, name):
self.saveWorkspaces()
try:
workspace = self.getWorkspaceManager().openWorkspace(name)
self.getModelController().setWorkspace(workspace)
except Exception:
model.api.log(
"An exception was captured while opening \
workspace %s\n%s" % (name, traceback.format_exc()), "ERROR")
class GuiApp(qt.QApplication, FaradayUi):
def __init__(self, model_controller, plugin_manager, workspace_manager):
FaradayUi.__init__(self,
model_controller,
plugin_manager,
workspace_manager)
qt.QApplication.__init__(self, [])
self._shell_envs = dict()
model.guiapi.setMainApp(self)
self._main_window = MainWindow(CONF.getAppname(),
self,
self.getModelController(),
self.getPluginManager())
self.setMainWidget(self.getMainWindow())
notifier = model.log.getNotifier()
notifier.widget = self._main_window
model.guiapi.notification_center.registerWidget(self._main_window)
self._splash_screen = qt.QSplashScreen(
qt.QPixmap(os.path.join(CONF.getImagePath(), "splash2.png")),
qt.Qt.WStyle_StaysOnTop)
def getMainWindow(self):
return self._main_window
def run(self, args):
self._main_window.createShellTab()
self.createLoggerWidget()
self._main_window.showAll()
exit_code = self.exec_loop()
return exit_code
def createLoggerWidget(self):
if not model.log.getLogger().isGUIOutputRegistered():
model.log.getLogger().registerGUIOutput(self._main_window.getLogConsole())
def loadWorkspaces(self):
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
splash_timer = qt.QTimer.singleShot(1700, lambda *args: None)
self._splash_screen.show()
def stopSplashScreen(self):
self._splash_screen.finish(self._main_window)
def quit(self):
model.log.getLogger().clearWidgets()
self.getMainWindow().hide()
envs = [env for env in self._shell_envs.itervalues()]
for env in envs:
env.terminate()
# exit status
notifier = model.log.getNotifier()
notifier.widget = None
qt.QApplication.quit(self)
def postEvent(self, receiver, event):
if receiver is None:
receiver = self.getMainWindow()
qt.QApplication.postEvent(receiver, QtCustomEvent.create(event))
def createShellEnvironment(self, name=None):
model.api.devlog("createShellEnvironment called \
- About to create new shell env with name %s" % name)
shell_env = ShellEnvironment(name, self,
self.getMainWindow().getTabManager(),
self.model_controller,
self.plugin_manager.createController,
self.deleteShellEnvironment)
self._shell_envs[name] = shell_env
self.getMainWindow().addShell(shell_env.widget)
shell_env.run()
def deleteShellEnvironment(self, name, ref=None):
def _closeShellEnv(name):
try:
env = self._shell_envs[name]
env.terminate()
tabmanager.removeView(env.widget)
del self._shell_envs[name]
except Exception:
model.api.devlog("ShellEnvironment could not be deleted")
model.api.devlog("%s" % traceback.format_exc())
model.api.devlog("deleteShellEnvironment called \
- name = %s - ref = %r" % (name, ref))
tabmanager = self.getMainWindow().getTabManager()
if len(self._shell_envs) > 1:
_closeShellEnv(name)
else:
if ref is not None:
result = self.getMainWindow().exitFaraday()
if result == qt.QDialog.Accepted:
self.quit()
else:
_closeShellEnv(name)
self.getMainWindow().createShellTab()
def removeWorkspace(self, name):
model.api.log("Removing Workspace: %s" % name)
return self.getWorkspaceManager().removeWorkspace(name)
def syncWorkspaces(self):
try:
self.getWorkspaceManager().saveWorkspaces()
except Exception:
model.api.log("An exception was captured while synchronizing \
workspaces\n%s" % traceback.format_exc(), "ERROR")
def saveWorkspaces(self):
try:
self.getWorkspaceManager().saveWorkspaces()
except Exception:
model.api.log("An exception was captured while saving \
workspaces\n%s" % traceback.format_exc(), "ERROR")
def createWorkspace(self, name, description="", w_type=""):
if name in self.getWorkspaceManager().getWorkspacesNames():
model.api.log("A workspace with name %s already exists"
% name, "ERROR")
else:
model.api.log("Creating workspace '%s'" % name)
model.api.devlog("Looking for the delegation class")
manager = self.getWorkspaceManager()
workingClass = None
if w_type and w_type in globals():
# If set as argument, otherwise let creation delegate behaviour
workingClass = globals()[w_type]
w = manager.createWorkspace(name, description, workspaceClass=workingClass)
self.getWorkspaceManager().setActiveWorkspace(w)
self.getModelController().setWorkspace(w)
self.getMainWindow().refreshWorkspaceTreeView()
self.getMainWindow().getWorkspaceTreeView().loadAllWorkspaces()
def openWorkspace(self, name):
self.saveWorkspaces()
try:
workspace = self.getWorkspaceManager().openWorkspace(name)
self.getModelController().setWorkspace(workspace)
except Exception:
model.api.log("An exception was captured while opening \
workspace %s\n%s" % (name, traceback.format_exc()), "ERROR")
