使用ruamel更新Yaml文件以及新字段

时间:2019-04-18 11:31:23

标签: yaml pyyaml ruamel.yaml

我正在尝试使用ruamel python更新yaml文件。

proc=subprocess.Popen(['kubectl','get','pod','web3','-o','yaml','--export'], stdout=subprocess.PIPE)
rein=proc.stdout.read()
result, indent, block_seq_indent = ruamel.yaml.util.load_yaml_guess_indent(rein, preserve_quotes=True)

到目前为止,我已经尝试过:

result['spec'].append('nodeSelector')

给出错误:

result['spec'].append('nodeSelector')
AttributeError: 'CommentedMap' object has no attribute 'append'

也尝试过这样:

result['spec']['nodeSelector']['kubernetes.io/hostname']='kubew1'

给予:

result['spec']['nodeSelector']['kubernetes.io/hostname']='kubew1'
File "/usr/local/lib/python3.6/dist-packages/ruamel/yaml/comments.py", line 752, in __getitem__
return ordereddict.__getitem__(self, key)
KeyError: 'nodeSelector'

我的Intial Yaml文件是:

apiVersion: v1
kind: Pod
metadata:
  creationTimestamp: null
  labels:
    app: demo
    name: web
  name: web3
  selfLink: /api/v1/namespaces/default/pods/web3
spec:
  containers:
  - image: aexlab/flask-sample-one
    imagePullPolicy: Always
    name: web
    ports:
    - containerPort: 5000
      name: http
      protocol: TCP
    resources: {}
    terminationMessagePath: /dev/termination-log
    terminationMessagePolicy: File
    volumeMounts:
    - mountPath: /var/run/secrets/kubernetes.io/serviceaccount
      name: default-token-7bcc9
      readOnly: true
  dnsPolicy: ClusterFirst
  enableServiceLinks: true

我想在'spec'中添加的期望字段是:

  nodeSelector:
    kubernetes.io/hostname: kubew1

任何想法如何通过ruamel库实现这一目标。

1 个答案:

答案 0 :(得分:1)

在YAML文件中,您的根级别集合是一个映射,并且该映射中的键spec的值本身就是一个映射。这两个映射都使用名为dict的{​​{1}}作为类似ruamel.yaml的对象加载。

与普通CommentedMap一样,您可以添加键值对,已删除键(及其值)以及更新键的值,但是没有dict方法,就像列表(即在列表中添加额外的项目)。

您的输出有点简洁,但是您当然不能只将.append()添加到任何内容(列表/序列或字典/映射),并期望添加nodeSelector(本身具有映射关系) )。

您的尝试:

kubernetes.io/hostname: kubew1

无法工作,因为没有字典result['spec']['nodeSelector']['kubernetes.io/hostname'] = 'kubew1' ,您无法在其中添加密钥result['spec']['nodeSelector']

您要么首先必须创建一个以emtpy dict作为值的键:

kubernetes.io/hostname

或做

result['spec']['nodeSelector'] = {}
result['spec']['nodeSelector']['kubernetes.io/hostname'] = 'kubew1'

请注意,以上内容与result['spec']['nodeSelector'] = {'kubernetes.io/hostname']: 'kubew1'} 没有多大关系,这只是基本的Python数据结构操作。还要注意,ruamel名称空间中有100多个库,其中ruamel.yaml只是作为开源发布的几个库之一,因此使用ruamel.yaml并不是很明确的声明,尽管当然上下文经常提供有关您实际使用哪个库的足够信息。