递归函数用于读取 YAML 文件，查找包含其他 YAML 文件的键。

huangapple
go评论87阅读模式
英文:

Recursive function to read yaml files looking for a key which includes other yaml files

问题

我正在尝试编写一个函数，该函数读取一个 YAML 文件并查找特定的键。如果该键不存在，该函数应该检查是否存在 "_include" 键，然后检查该 YAML 文件，直到所有选项都尝试完毕或找到该键。

  1. import yaml
  2. import os.path
  4. def load_yaml_data(yaml_path):
  5.     """读取 YAML 文件并返回数据"""
  6.     try:
  7.         with open(yaml_path, 'r', encoding="utf-8") as yaml_file:
  8.             yaml_data = yaml.safe_load(yaml_file)
  9.             return yaml_data
  10.     except FileNotFoundError:
  11.         return None
  12.     except yaml.YAMLError as exc:
  13.         return None
  15. def find_key_in_yaml_recursive(yaml_path, key):
  16.     print(f"打开 {yaml_path}")
  17.     yaml_data = load_yaml_data(yaml_path)
  18.     if yaml_data:
  19.         if key in yaml_data:
  20.             print(f"{key} 在 {yaml_path} 中找到")
  21.             return yaml_path
  22.         elif '_include' in yaml_data:
  23.             for include in yaml_data['_include']:
  24.                 include_path = os.path.expandvars(include)
  25.                 if not os.path.isabs(include_path):
  26.                     include_path = os.path.abspath('/'.join([os.path.dirname(yaml_path), include_path]))
  27.                 result = find_key_in_yaml_recursive(include_path, key)
  28.                 if result:
  29.                     return result
  31. if __name__ == "__main__":
  32.     print(find_key_in_yaml_recursive("./test.yaml", "foo"))

目录结构:

  1. .
  2. |    test.yaml
  3. └─── test
  4. │   │    test.yaml
  5. │   └─── deeper
  6. │       │   test.yaml
  7. │       │   notexist.yaml

./test.yaml

  1. ---
  2. _include:
  3.   - "test/test.yaml"
  4. not_foo:
  5.   - "some data"

./test/test.yaml

  1. ---
  2. _include:
  3.   - ./deeper/notexist.yaml
  4.   - ./deeper/test.yaml
  5. foop:
  6.   - "somedata"

./test/deeper/notexist.yaml

  1. ---
  2. nothinguseful:
  3.   - data

./test/deeper/test.yaml

  1. ---
  2. foo:
  3.  - some data
  4. somethingelse:
  5.   - data

'foo' 存在于 ./test/deeper/test.yaml 中。在大多数情况下，该函数将找到它并返回完整的 ./test/deeper/test.yaml 路径。但是，当 ./test/test.yaml 具有一个列出 deeper/notexist.yaml 在 deeper/test.yaml 之前的 "_include" 数组时，该函数返回 None。

我并没有进行太多的递归编程，但我知道你需要一个基本情况，但不确定如何在这种情况下实现基本情况。

英文:

I am attempting to write a function that reads a yaml file looking for a certain key. if the key does not exist, the function should check for an '_include' key and then check that yaml file for the same key until all options are exhausted or it finds the key.

  1. import yaml
  2. import os.path
  4. def load_yaml_data(yaml_path):
  5.     """Read a YAML file and return the data"""
  6.     try:
  7.         with open(yaml_path, 'r', encoding="utf-8") as yaml_file:
  8.             yaml_data = yaml.safe_load(yaml_file)
  9.             return yaml_data
  10.     except FileNotFoundError:
  11.         return None
  12.     except yaml.YAMLError as exc:
  13.         return None
  15. def find_key_in_yaml_recursive(yaml_path, key):
  16.     print(f"opening {yaml_path}")
  17.     yaml_data = load_yaml_data(yaml_path)
  18.     # print(f"Yaml data : {yaml_data}")
  19.     if yaml_data:
  20.         if key in yaml_data:
  21.             print(f"{key} found in {yaml_path}")
  22.             return yaml_path
  23.         elif '_include' in yaml_data:
  24.             for include in yaml_data['_include']:
  25.                 include_path = os.path.expandvars(include)
  26.                 if not os.path.isabs(include_path):
  27.                     include_path = os.path.abspath('/'.join([os.path.dirname(yaml_path), include_path]))
  28.                 return find_key_in_yaml_recursive(include_path, key)
  30. if __name__ == "__main__":
  31.     print(find_key_in_yaml_recursive("./test.yaml", "foo"))

Dir structure:

  1. .
  2. |    test.yaml
  3. └─── test
  4. │   │    test.yaml
  5. │   └─── deeper
  6. │       │   test.yaml
  7. │       │   notexist.yaml

./test.yaml

  1. ---
  2. _include:
  3.   - "test/test.yaml"
  4. not_foo:
  5.   - "some data"

./test/test.yaml

  1. ---
  2. _include:
  3.   - ./deeper/notexist.yaml
  4.   - ./deeper/test.yaml
  5. foop:
  6.   - "somedata"

./test/deeper/notexist.yaml

  1. ---
  2. nothinguseful:
  3.   - data

./test/deeper/test.yaml

  1. ---
  2. foo:
  3.  - some data
  4. somethingelse:
  5.   - data

'foo' exists in ./test/deeper/test.yaml. In most circumstances, the function will find it and return the full ./test/deeper/test.yaml path. However, when ./test/test.yaml has an '_include' array that lists deeper/notexist.yaml before deeper/test.yaml the function returns None.
I have not done much recursive programming but i know you need a base case but i am unsure how to implement a base case in this scenario.

答案1

得分: 1

我认为这应该可以工作：

  1. import yaml
  2. import os.path
  4. def load_yaml_data(yaml_path):
  5.     """读取一个YAML文件并返回数据"""
  6.     try:
  7.         with open(yaml_path, 'r', encoding="utf-8") as yaml_file:
  8.             yaml_data = yaml.safe_load(yaml_file)
  9.             return yaml_data
  10.     except FileNotFoundError:
  11.         return None
  12.     except yaml.YAMLError as exc:
  13.         return None
  15. def find_key_in_yaml_recursive(yaml_path, key):
  16.     print(f"打开 {yaml_path}")
  17.     yaml_data = load_yaml_data(yaml_path)
  18.     # print(f"Yaml数据 : {yaml_data}")
  19.     if yaml_data:
  20.         if key in yaml_data:
  21.             print(f"{key} 在 {yaml_path} 中找到")
  22.             return yaml_path
  23.         elif '_include' in yaml_data:
  24.             for include in yaml_data['_include']:
  25.                 include_path = os.path.expandvars(include)
  26.                 if not os.path.isabs(include_path):
  27.                     include_path = os.path.abspath('/'.join([os.path.dirname(yaml_path), include_path]))
  28.                 var = find_key_in_yaml_recursive(include_path, key)
  29.                 if var:
  30.                     return var
  31.     return None
  33. if __name__ == "__main__":
  34.     print(find_key_in_yaml_recursive("./test.yaml", "foo"))
英文:

I think this should work:

  1. import yaml
  2. import os.path
  4. def load_yaml_data(yaml_path):
  5.     """Read a YAML file and return the data"""
  6.     try:
  7.         with open(yaml_path, 'r', encoding="utf-8") as yaml_file:
  8.             yaml_data = yaml.safe_load(yaml_file)
  9.             return yaml_data
  10.     except FileNotFoundError:
  11.         return None
  12.     except yaml.YAMLError as exc:
  13.         return None
  15. def find_key_in_yaml_recursive(yaml_path, key):
  16.     print(f"opening {yaml_path}")
  17.     yaml_data = load_yaml_data(yaml_path)
  18.     #print(f"Yaml data : {yaml_data}")
  19.     if yaml_data:
  20.         if key in yaml_data:
  21.             print(f"{key} found in {yaml_path}")
  22.             return yaml_path
  23.         elif '_include' in yaml_data:
  24.             for include in yaml_data['_include']:
  25.                 include_path = os.path.expandvars(include)
  26.                 if not os.path.isabs(include_path):
  27.                     include_path = os.path.abspath('/'.join([os.path.dirname(yaml_path), include_path]))
  28.                 var = find_key_in_yaml_recursive(include_path, key)
  29.                 if var:
  30.                     return var  
  31.     return None
  33. if __name__ == "__main__":
  34.     print(find_key_in_yaml_recursive("./test.yaml", "foo"))

huangapple
  • 本文由 发表于 2023年3月7日 21:59:10
  • 转载请务必保留本文链接：https://go.coder-hub.com/75662943.html
  • python
  • recursion
  • yaml
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定