|
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/") |
|
|