INNER CODE UNIT · Python
generate_csv
magicsword-io/LOLRMM · bin/generate_domains_csv.py:72
def generate_csv(yaml_dir, output_file):
"""Generate a CSV file with domains from all YAML files."""
all_domains = []
# Find all YAML files in the directory
yaml_files = sorted(glob.glob(os.path.join(yaml_dir, "*.y*ml")))
print(f"Found {len(yaml_files)} YAML files in {yaml_dir}")
# Extract domains from each YAML file
for yaml_file in yaml_files:
domains = extract_domains(yaml_file)
all_domains.extend(domains)
unique_domains = []
seen = set()
for domain, tool in all_domains:
key = (domain, tool)
if key in seen: