Download scripts/compile_safe_codes_to_json.py from lilbool/vuln-code-analysis: direct link, hf CLI and curl.
- Browser
- Download file 1.83 kB
-
https://huggingface.co/lilbool/vuln-code-analysis/resolve/main/scripts/compile_safe_codes_to_json.py
- Command line
-
hf download hf://lilbool/vuln-code-analysis/scripts/compile_safe_codes_to_json.py
-
curl -L -o compile_safe_codes_to_json.py https://huggingface.co/lilbool/vuln-code-analysis/resolve/main/scripts/compile_safe_codes_to_json.py
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) | |