def get_white_list(datasets_file_path: str, dataset_key: Optional[str]) -> List[str]:
if dataset_key is None:
return []
datasets = read_yaml(datasets_file_path)
if dataset_key not in datasets:
raise ValueError("Invalid dataset: '{}'".format(dataset_key))
return datasets[dataset_key]
def __load_data(run_file_path: str):
data = {
"result": None,
"runtime": None,
"message": "",
"md5": None
}
data.update(read_yaml(run_file_path) if exists(run_file_path) else {})
return data
def __init__(self, run_mode: DetectorMode, detector: Detector, version: ProjectVersion, findings_base_path: str,
findings_filter: FindingsFilter):
self.run_mode = run_mode
self.detector = detector
self.version = version
self._findings_base_path = findings_base_path
self._findings_file_path = join(self._get_findings_path(), self.FINDINGS_FILE)
self.__FINDINGS = None
self.__POTENTIAL_HITS = None
data = {
"result": None,
"runtime": None,
"message": "",
"md5": None
}
data.update(read_yaml(self._run_file_path) if exists(self._run_file_path) else {})
self.result = Result[data["result"]] if data["result"] else None
self.runtime = data["runtime"]
self.message = data["message"]
self._detector_md5 = data["md5"]
self.findings_filter = findings_filter
