python Python - 在YAML列表中搜索要更改或插入的值

Posted 2021-05-09

#!/usr/bin/python

import yaml, sys

if len(sys.argv) < 3:
    print 'Usage: run.py <inputfile> <outputfile>'
    print '<inputfile>: connector.yml'
    print '<outputfile>: userConfigurations.yaml'
    sys.exit()

key_name = 'packageId'
list_name = 'packages'


class MyDumper(yaml.Dumper):
    """
    json list indenter fix
    """
    def increase_indent(self, flow=False, indentless=False):
        return super(MyDumper, self).increase_indent(flow, False)


output_file = sys.argv[2]
input_file = sys.argv[1]


x = yaml.load(open(output_file))
y = yaml.load(open(input_file))
for i, element in enumerate(x[list_name]):
    if element[key_name] == y[0][key_name]:
        x[list_name].pop(i)
x[list_name].append(y[0])

f = open(output_file, 'w')
f.write(yaml.dump(x, Dumper=MyDumper, default_flow_style=False))
f.close()