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_inspect_task_runner(self, scan_task): """Create inspection TaskRunner using source_type.""" source_type = scan_task.source.source_type runner = None if source_type == Source.NETWORK_SOURCE_TYPE: runner = network.InspectTaskRunner(self.scan_job, scan_task) elif source_type == Source.VCENTER_SOURCE_TYPE: runner = vcenter.InspectTaskRunner(self.scan_job, scan_task) elif source_type == Source.SATELLITE_SOURCE_TYPE: runner = satellite.InspectTaskRunner(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