使用python3编辑yaml参数

Question

我有一个yaml文件的列表，我想以同样的方式进行编辑。将以下代码块放在每个文件的spec.template.imagePullSecret和spec.template.container.imagePullPolicy下：

imagePullSecrets: 
  - dockerpullsecret
container:
  imagePullPolicy: IfNotPresent

这是一个文件的示例：

apiVersion: argoproj.io/v1alpha1
kind: Sensor
meta data:
  name: sensor-finanda-ci
  namespace: argo-events
spec:  
  template:
    serviceAccountName: argo-events-sa
  dependencies:
    - name: eventsource-iv
      eventSourceName: eventsource-iv
      eventName: iv

这是我想要的输出：

apiVersion: argoproj.io/v1alpha1
kind: Sensor
meta data:
  name: sensor-finanda-ci
  namespace: argo-events
spec:  
  template:
    serviceAccountName: argo-events-sa
    imagePullSecrets: 
      - dockerpullsecret
    container:
      imagePullPolicy: IfNotPresent
  dependencies:
    - name: eventsource-iv
      eventSourceName: eventsource-iv
      eventName: iv

Answer 1

如果你一心想要一个Python解决方案，那么下面这样的代码可能就行了：

import os
import yaml

for yaml_filename in [filename for filename in os.listdir('.') if filename.endswith('.yaml')]:
    with open(yaml_filename, 'r') as yaml_file:
        yaml_obj = yaml.safe_load(yaml_file)

    if 'spec' not in yaml_obj:
        yaml_obj['spec'] = {"template": {}}
    if 'template' not in yaml_obj['spec']:
        yaml_obj['spec']['template'] = {"imagePullSecrets": [], "container": {}}
    if 'container' not in yaml_obj['spec']['template']:
        yaml_obj['spec']['template']['container'] = {}

    yaml_obj['spec']['template']['imagePullSecrets'] = ["dockerpullsecret"]
    yaml_obj['spec']['template']['container']['imagePullPolicy'] = 'IfNotPresent'

    with open(yaml_filename, 'w') as yaml_file:
        yaml.dump(yaml_obj, yaml_file)

如果您对命令行实用程序持开放态度，这将更加简洁：

循环遍历文件并使用yq就地编辑它们。

for file in *.yaml; do 
  yq -i e '.spec.template.imagePullSecrets = ["dockerpullsecret"] | .spec.template.container.imagePullPolicy = "IfNotPresent"' "$file"
done