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