30 lines
822 B
Python
30 lines
822 B
Python
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/")
|