| import os | |
| import yaml | |
| def convert_py_to_yaml(py_file_path): | |
| with open(py_file_path, encoding="utf-8") as py_file: | |
| py_content = py_file.read() | |
| local_vars = {} | |
| exec(py_content, {}, local_vars) | |
| yaml_file_path = os.path.splitext(py_file_path)[0] + ".yaml" | |
| with open(yaml_file_path, "w", encoding="utf-8") as yaml_file: | |
| yaml.dump(local_vars, yaml_file, default_flow_style=False, allow_unicode=True) | |
| def process_directory(path): | |
| for root, dirs, files in os.walk(path): | |
| for filename in files: | |
| if filename.endswith(".py"): | |
| py_file_path = os.path.join(root, filename) | |
| convert_py_to_yaml(py_file_path) | |
| print(f"convert {py_file_path} to YAML format") | |
| if __name__ == "__main__": | |
| process_directory("../configs/") | |