vuln-code-analysis / scripts /compile_safe_codes_to_json.py
lilbool's picture
Upload 212 files
497f2f3 verified
raw
history blame
1.83 kB
import os
import json
import re
def sanitize_filename(filename):
"""Remove ou substitui caracteres inv谩lidos em nomes de arquivos."""
return re.sub(r'[<>:"/\\|?*]', '_', filename)
def compile_safe_codes_to_json(input_dir, output_file):
"""Compila todos os arquivos de c贸digos seguros em um 煤nico JSON."""
compiled_data = []
# Iterar por todos os arquivos na pasta de c贸digos seguros
for filename in os.listdir(input_dir):
if filename.endswith(".txt"): # Filtrar apenas arquivos de texto
file_path = os.path.join(input_dir, filename)
try:
with open(file_path, 'r', encoding='utf-8') as f:
content = f.read()
# Adicionar os dados ao JSON
compiled_data.append({
"filename": sanitize_filename(filename),
"content": content,
"safe": True # Marcar como c贸digo seguro
})
except Exception as e:
print(f"[ERROR] Could not process file {filename}: {e}")
# Salvar os dados compilados no arquivo JSON
try:
with open(output_file, 'w', encoding='utf-8') as json_file:
json.dump(compiled_data, json_file, indent=4)
print(f"[SUCCESS] Compiled {len(compiled_data)} safe codes into {output_file}")
except Exception as e:
print(f"[ERROR] Could not write JSON file: {e}")
if __name__ == "__main__":
input_directory = "safe-code-analyzer/safe_codes" # Diret贸rio dos c贸digos seguros
output_json_file = "safe-code-analyzer/compiled_safe_codes.json" # Arquivo JSON de sa铆da
os.makedirs(os.path.dirname(output_json_file), exist_ok=True)
compile_safe_codes_to_json(input_directory, output_json_file)