Добавить настраиваемые поля в файл JSON на основе пользовательского ввода из файла YAML - Python3

Question

Я новичок в Python.У меня есть один файл YAML, и я обращаюсь к нему с помощью файла Python.В файле YAML есть опция полей.В файле YAML Пользователь может устанавливать переменные со значениями.Файл Python считывает переменные со значениями, а затем добавляет их в файл JSON.Обратите внимание, что переменные и значения могут быть изменены в соответствии с пользователем.
Как я могу сделать это возможным?

Вот пример кода:

import yaml
from datetime import datetime
import os
import json

#name for json file
name = "stack.json"

#load data from yml file
data = yaml.safe_load(open('stack.yml'))
data2 = data.get('heartbeat.monitors')

#Current time stamp
timestamp = datetime.now().strftime("%B %d %Y, %H:%M:%S")

#ip
ip ='192.168.1.1'

#getting data from the field and assign it to variable
for item in data2:
    if item["type"] == "icmp":
        fields_under_root = (item["fields_under_root"])

        # if fields_under_root is true,code goes here
        if fields_under_root == True:
            fields = (item["fields"])
            print(fields)
            locals().update(fields)
        #code to be entered

        #if fields_under_root is false, code goes here
        elif fields_under_root == False:
            fields = (item["fields"])
            print(fields)
        #code to be entered
#For writing in JSON File
#Creates a JSON file if not exists
if not os.path.exists(name):
    with open(name, 'w') as f:
        f.write('{}')

#list for storing the values
result = [(timestamp, {'monitor.ip': ip,"fields": fields })]

#For writing in JSON File
with open(name, 'rb+') as f:
    f.seek(-1, os.SEEK_END)
    f.truncate()
    for entry in result:
        _entry = '"{}":{},\n'.format(entry[0], json.dumps(entry[1]))
        _entry = _entry.encode()
        f.write(_entry)
    f.write('}'.encode('ascii'))

В файле YAML:

heartbeat.monitors:
- type: icmp
  fields:
    a: steven
    b: kumar

  fields_under_root: True

Мой вывод в JSON-файле:

{"February 18 2019, 17:04:30":{"monitor.ip": "192.168.1.1", "fields": {"b": "kumar", "a": "steven"}},
}

Требуемый вывод, если fields_under_root равен True:

{"February 18 2019, 17:04:30":{"monitor.ip": "192.168.1.1", "b": "kumar", "a": "steven"},
}

Требуемый вывод, если fields_under_root равен False:

{"February 18 2019, 17:04:30":{"monitor.ip": "192.168.1.1", "fields.b": "kumar", "fields.a": "steven"},
}

Answer 1

Учитывая входной файл:

...

heartbeat.monitors:
- type: icmp
  fields:
    a: steven
    b: kumar
  # If this option is set to true, the custom fields are stored as top-level
  # fields in the output document instead of being grouped under a fields
  # sub-dictionary. Default is false.
  fields_under_root: True

ваша программа должна иметь функцию, которая выполняет всю работу, тогда легко протестировать обе версии ввода, обновив входной файл .:

import ruamel.yaml
from datetime import datetime
import os
import json

# name for json file
json_name = 'stack.json'
yaml_name = 'stack.yaml'


# Current time stamp
timestamp = datetime.now().strftime('%B %d %Y, %H:%M:%S')


def gen_output(data, json_filename):
    ip = '192.168.1.1'
    val = {'monitor.ip': ip}
    result = dict(timestamp=val)

    # getting data from the field and assign it to variable

    for item in data:
        if item['type'] == 'icmp':
            fields = item.pop('fields')
            if item['fields_under_root']:
                val.update(fields)
            else:
                nf = {}
                for k in fields:
                    nf['fields.' + k] = fields[k]
                val.update(nf)

    with open(json_filename, 'w') as fp:
        json.dump(result, fp, ensure_ascii=False)


# load data from YAML file
yaml = ruamel.yaml.YAML(typ='safe')
with open('stack.yaml') as fp:
    data = yaml.load(fp)
data2 = data.get('heartbeat.monitors')

gen_output(data2, json_name)

# show the output file
with open('stack.json') as fp:
    print(fp.read())

print('---------')
# update the YAML input file
with open('stack.yaml') as fp:
    yamlrt = ruamel.yaml.YAML() # default is round-trip, preserving comments
    data = yaml.load(fp)
data['heartbeat.monitors'][0]['fields_under_root'] = False
with open(yaml_name, 'wb') as fp:
    yamlrt.dump(data, fp)

with open('stack.yaml') as fp:
    data = yaml.load(fp)
data2 = data.get('heartbeat.monitors')

gen_output(data2, json_name)
with open('stack.json') as fp:
    print(fp.read())

, что дает:

{"timestamp": {"monitor.ip": "192.168.1.1", "a": "steven", "b": "kumar"}}
---------
{"timestamp": {"monitor.ip": "192.168.1.1", "fields.a": "steven", "fields.b": "kumar"}}

Общие комментарии:

• не проверять на == False или == True, если вообщепроверить с помощью is False или is True, но лучше просто проверить переменную x, которая содержит логическое значение с if x:, и не делать оба, логическое значение, если не True, это False, поэтому используйтеобычный else:

• не волнуйтесь с выводом JSON.Не нужно усекать, искать и т. Д. Просто убедитесь, что ваши данные правильно сформированы, а затем сбросьте за один раз.Это гарантирует, что ваш вывод действителен в формате JSON.

• не добавляйте комментарии с указанием очевидного #ip

• добавьте пробел после комментарияtoken #, многие владельцы жалоб жалуются на отсутствие этого

• Я использую ruamel.yaml в приведенном выше (отказ от ответственности: я являюсь автором этого пакета), потому что с PyYAML программноОбновление файла YAML приведет к потере вами информации о комментариях.И поскольку YAML(typ='safe').load() быстрее, чем yaml.load() в PyYAML (кроме этого PyYAML только поддерживает устаревший YAML 1.1)