Files
WithSecureLabs-lolcerts/scripts/generate_csv.py
T
2024-02-09 02:28:20 -06:00

33 lines
1.1 KiB
Python

import glob
import yaml
for folder in ("leaked", "malicious"):
yml_files = glob.glob(f"../{folder}/*.yml", recursive=True)
csv_content = "name,status,source,description,references,date,author,issuer,timestamp,serial\n"
# Generate the csv file for each folder
for yml_file in yml_files:
# Load the yml file
with open(yml_file, 'r') as stream:
try:
yml = yaml.safe_load(stream)
except yaml.YAMLError as exc:
print(exc)
for serial in yml["serial"]:
# Generate the csv entry
entry = ",".join([
yml["name"],
yml["meta"]["status"],
yml["meta"]["source"],
yml["meta"]["description"],
yml["meta"]["references"],
yml["meta"]["date"],
yml["meta"]["author"],
yml["issuer"],
str(yml["timestamp"]) if "timestamp" in yml else "",
serial
])
csv_content += f"{entry}\n"
with open(f"../csv/{folder}.csv", "w") as f:
f.write(csv_content)