def _create_task_runner(self, scan_task):
"""Create ScanTaskRunner using scan_type and source_type."""
scan_type = scan_task.scan_type
source_type = scan_task.source.source_type
runner = None
if (scan_type == ScanTask.SCAN_TYPE_CONNECT and
source_type == Source.NETWORK_SOURCE_TYPE):
runner = network.ConnectTaskRunner(
self.scan_job, scan_task)
elif (scan_type == ScanTask.SCAN_TYPE_CONNECT and
source_type == Source.VCENTER_SOURCE_TYPE):
runner = vcenter.ConnectTaskRunner(
self.scan_job, scan_task)
elif (scan_type == ScanTask.SCAN_TYPE_CONNECT and
source_type == Source.SATELLITE_SOURCE_TYPE):
runner = satellite.ConnectTaskRunner(
self.scan_job, scan_task)
elif (scan_type == ScanTask.SCAN_TYPE_INSPECT and
source_type == Source.NETWORK_SOURCE_TYPE):
runner = network.InspectTaskRunner(
self.scan_job, scan_task)
elif (scan_type == ScanTask.SCAN_TYPE_INSPECT and
source_type == Source.VCENTER_SOURCE_TYPE):
runner = vcenter.InspectTaskRunner(
self.scan_job, scan_task)
elif (scan_type == ScanTask.SCAN_TYPE_INSPECT and
source_type == Source.SATELLITE_SOURCE_TYPE):
runner = satellite.InspectTaskRunner(
self.scan_job, scan_task)
return runner
def _create_connect_task_runner(self, scan_task):
"""Create connection TaskRunner using source_type."""
source_type = scan_task.source.source_type
runner = None
if source_type == Source.NETWORK_SOURCE_TYPE:
runner = network.ConnectTaskRunner(self.scan_job, scan_task)
elif source_type == Source.VCENTER_SOURCE_TYPE:
runner = vcenter.ConnectTaskRunner(self.scan_job, scan_task)
elif source_type == Source.SATELLITE_SOURCE_TYPE:
runner = satellite.ConnectTaskRunner(self.scan_job, scan_task)
return runner
def create_task_runner(self, scan_task):
"""Create ScanTaskRunner using scan_type and source_type."""
scan_type = scan_task.scan_type
source_type = scan_task.source.source_type
runner = None
if (scan_type == ScanTask.SCAN_TYPE_CONNECT
and source_type == Source.NETWORK_SOURCE_TYPE):
runner = network.ConnectTaskRunner(self.scan_job, scan_task,
self.conn_results)
elif (scan_type == ScanTask.SCAN_TYPE_CONNECT
and source_type == Source.VCENTER_SOURCE_TYPE):
runner = vcenter.ConnectTaskRunner(self.scan_job, scan_task,
self.conn_results)
elif (scan_type == ScanTask.SCAN_TYPE_INSPECT
and source_type == Source.NETWORK_SOURCE_TYPE):
runner = network.InspectTaskRunner(self.scan_job, scan_task,
self.inspect_results)
elif (scan_type == ScanTask.SCAN_TYPE_INSPECT
and source_type == Source.VCENTER_SOURCE_TYPE):
runner = vcenter.InspectTaskRunner(self.scan_job, scan_task,
self.inspect_results)
return runner