def process_workspaces(mappers_manager, plugin_manager, query, disable_polling):
report_managers = []
controllers = []
for workspace in query.all():
pending_actions = Queue()
plugin_controller = PluginController(
'PluginController',
plugin_manager,
mappers_manager,
pending_actions
)
mappers_manager.createMappers(workspace.name)
controller = ModelController(mappers_manager, pending_actions)
workspace_manager = WorkspaceManager(mappers_manager)
setUpAPIs(controller, workspace_manager, hostname=None, port=None)
controller.start()
controllers.append(controller)
report_manager = ReportManager(
0.1,
workspace.name,
plugin_controller,
polling=not disable_polling
)
report_managers.append(report_manager)
report_manager.start()
for report_manager in report_managers:
report_manager.stop()
for controller in controllers:
controller.stop()
class FaradayUi(object):
def __init__(self, model_controller, plugin_manager,
workspace_manager, plugin_controller, gui="gtk"):
self.model_controller = model_controller
self.plugin_manager = plugin_manager
self.workspace_manager = workspace_manager
self.plugin_controller = plugin_controller
self.report_manager = None
def getModelController(self):
return self.model_controller
def getPluginManager(self):
return self.plugin_manager
def getWorkspaceManager(self):
return self.workspace_manager
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
pass
def stopSplashScreen(self):
pass
def splashMessage(self, message):
pass
def loadWorkspaces(self):
pass
def run(self, args):
pass
def quit(self):
pass
def postEvent(self, receiver, event):
pass
def createLoggerWidget(self):
pass
def openWorkspace(self, name):
"""Open a workspace by name. Returns the workspace of raises an
exception if for some reason it couldn't.
"""
if self.report_manager:
self.report_manager.stop()
self.report_manager.join()
try:
ws = self.getWorkspaceManager().openWorkspace(name)
self.report_manager = ReportManager(
10,
name,
self.plugin_controller
)
self.report_manager.start()
except Exception as e:
raise e
return ws
class FaradayUi(object):
def __init__(self, model_controller=None, plugin_manager=None,
workspace_manager=None, gui="qt3"):
self.model_controller = model_controller
self.plugin_manager = plugin_manager
self.workspace_manager = workspace_manager
self.report_manager = None
def getModelController(self):
return self.model_controller
def getPluginManager(self):
return self.plugin_manager
def getWorkspaceManager(self):
return self.workspace_manager
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
pass
def stopSplashScreen(self):
pass
def splashMessage(self, message):
pass
def loadWorkspaces(self):
pass
def run(self, args):
pass
def quit(self):
pass
def postEvent(self, receiver, event):
pass
def createLoggerWidget(self):
pass
def openWorkspace(self, name):
# The method openWorkspace can return a workspace or
# raise en Exception. For now, just raise it to the caller
if self.report_manager:
self.report_manager.stop()
self.report_manager.join()
try:
ws = self.getWorkspaceManager().openWorkspace(name)
self.report_manager = ReportManager(
10, name)
self.report_manager.start()
except Exception as e:
raise e
return ws
def openDefaultWorkspace(self):
"""
Opens the default workspace (called 'untitled').
This method shouldn't fail, since the default workspace
should be always available
Returns the default workspace
"""
if self.report_manager:
self.report_manager.stop()
self.report_manager.join()
ws = self.getWorkspaceManager().openDefaultWorkspace()
self.report_manager = ReportManager(
10, ws.name)
self.report_manager.start()
return ws
class FaradayUi(object):
def __init__(self,
model_controller,
plugin_manager,
workspace_manager,
plugin_controller,
gui="qt3"):
self.model_controller = model_controller
self.plugin_manager = plugin_manager
self.workspace_manager = workspace_manager
self.plugin_controller = plugin_controller
self.report_manager = None
def getModelController(self):
return self.model_controller
def getPluginManager(self):
return self.plugin_manager
def getWorkspaceManager(self):
return self.workspace_manager
def setSplashImage(self, ipath):
pass
def startSplashScreen(self):
pass
def stopSplashScreen(self):
pass
def splashMessage(self, message):
pass
def loadWorkspaces(self):
pass
def run(self, args):
pass
def quit(self):
pass
def postEvent(self, receiver, event):
pass
def createLoggerWidget(self):
pass
def openWorkspace(self, name):
# The method openWorkspace can return a workspace or
# raise en Exception. For now, just raise it to the caller
if self.report_manager:
self.report_manager.stop()
self.report_manager.join()
try:
ws = self.getWorkspaceManager().openWorkspace(name)
self.report_manager = ReportManager(10, name,
self.plugin_controller)
self.report_manager.start()
except Exception as e:
raise e
return ws
def openDefaultWorkspace(self):
"""
Opens the default workspace (called 'untitled').
This method shouldn't fail, since the default workspace
should be always available
Returns the default workspace
"""
if self.report_manager:
self.report_manager.stop()
self.report_manager.join()
ws = self.getWorkspaceManager().openDefaultWorkspace()
self.report_manager = ReportManager(10, ws.name,
self.plugin_controller)
self.report_manager.start()
return ws
