如何用yaml中的字典替换字符串

Question

我有一个看起来像这样的yaml：

- k1: /etc/prometheus/auxiliaryfiles/mac-jwt-cne
  k2: abcd
  k3: 
  k4:
    z1:
     - asfa
  k5: zxcv
  key_001:
    __TARGETS__

我有一个词典，看起来像：

{'a':[12,23,34]}

想在yaml中用上面的字典替换__ targets __

key_001:
 - a:
   - 12
   - 23
   - 34

yaml_dict = readYaml("lib/above_Yaml.yaml")
d = {'a': [12, 23, 34]}
replacements = {'__TARGETS__':[d]}

for k, v in yaml_dict.items():
    if v in replacements:
        yaml_dict[k] = replacements[v]

dumped = yaml.safe_dump(yaml_dict)
print(dumped)

我 以下错误：

     if v in replacements:
 TypeError: unhashable type: 'dict'

Answer 1

我将使用递归搜索并替换的方法尝试：

from __future__ import annotations
from typing import Any

import yaml

yaml_str = r"""
  k1: /etc/prometheus/auxiliaryfiles/mac-jwt-cne
  k2: abcd
  k3:
  k4:
    z1:
     - asfa
  k5: zxcv
  key_001:
    __TARGETS__
"""


def recursive_repl(o: dict | list | Any, repl_dict: dict[str, Any]):
    """Recursively search and replace ``o`` with key-value pairs in ``repl_dict``"""
    if isinstance(o, dict):
        return {k: recursive_repl(v, repl_dict) for k, v in o.items()}

    if isinstance(o, list):
        return [recursive_repl(e, repl_dict) for e in o]

    return repl_dict.get(o, o)


d = {'a': [12, 23, 34]}
replacements = {'__TARGETS__': [d]}

# load yaml string into a dict
yaml_dict = yaml.safe_load(yaml_str)

# recursively search and replace with desired replacements
yaml_dict = recursive_repl(yaml_dict, replacements)

# convert back to a yaml string, and optionally write to file
dumped = yaml.safe_dump(yaml_dict)
print(dumped)

out：out：

k1: /etc/prometheus/auxiliaryfiles/mac-jwt-cne
k2: abcd
k3: null
k4:
  z1:
  - asfa
k5: zxcv
key_001:
- a:
  - 12
  - 23
  - 34

注意：这需要 pyyaml

pip install PyYAML