| import sys | |
| import yaml | |
| # read scanners from yaml file | |
| # return a list of scanners | |
| def read_scanners(path): | |
| scanners = [] | |
| with open(path, "r") as f: | |
| config = yaml.load(f, Loader=yaml.FullLoader) | |
| scanners = config.get("detectors", None) | |
| return scanners | |
| # convert a list of scanners to yaml file | |
| def write_scanners(scanners): | |
| with open("./scan_config.yaml", "w") as f: | |
| # save scanners to detectors in yaml | |
| yaml.dump({"detectors": scanners}, f) | |
| # convert column mapping dataframe to json | |
| def convert_column_mapping_to_json(df, label=""): | |
| column_mapping = {} | |
| column_mapping[label] = [] | |
| for _, row in df.iterrows(): | |
| column_mapping[label].append(row.tolist()) | |
| return column_mapping | |