Как отформатировать результат json массива?

Question

[evgeh4ik]

Есть данный код для парсинга xml:

#!/usr/bin/env python36

import os
import xml.dom.minidom
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-i', '--host' , help='Input hostname')
parser.add_argument('-z', '--zone' , help='Input zone name')
args = parser.parse_args()

spisok =[]
for rootDir in os.listdir(r'/mnt/'+args.host+'/'+args.zone):
        for root, dirs, files in os.walk(r'/mnt/'+args.host+'/'+args.zone+'/' + rootDir +'/defaultVars/Vars/Connections'):
            for file in files:
                if file.endswith(".substvar"):
                    path = os.path.join(root, file)
                    spisok.append(path)

for element in spisok:

	doc = xml.dom.minidom.parse(element)
	data ={}

	for info in doc.getElementsByTagName("globalVariable"):
		name = str(info.getElementsByTagName("name")[0].firstChild.data)
		value = str(info.getElementsByTagName("value")[0].firstChild.data)
		data[name]=value
	print(element,"\n",data,"\n")

На текущий момент вывод печатается в формате:

/mnt/'+args.host+'/'+args.zone+'/' + rootDir +'/defaultVars/Vars/Connections'/EMS/Data
{'Host': '10.64.73.137:1576', 'Login': 'ows_dbts', 'LoginTimeout': '10'}

Я хочу переделать вывод в вид:

{ rootDir:
 { EMS:
  { Host: 10.64.73.137:1576
    Login: ows_dbts
    LoginTimeout: 10
  }
 }
}

Каким образом корректно реализовать форматировние результата? Как выдергивать название директории ( в примере EMS, но директории могут меняться).

Answer 1

[Сергей Горностаев]

Как сохранять JSON не в минифицированном виде?

Comments

[Ivan Yakushenko]

На тостере напрашивается медаль "Одним ответом N вопросов"

[evgeh4ik]

Спасибо. А как можно вытащить название директории (в примере EMS)?

[Сергей Горностаев]

evgeh4ik, просто объединить циклы и задача станет совсем элементарной.

[evgeh4ik]

Сергей Горностаев, можно для более глупого человека на примере показать? :D

[Сергей Горностаев]

evgeh4ik, что-то вроде

for rootDir in os.listdir(r'/mnt/'+args.host+'/'+args.zone):
        for root, dirs, files in os.walk(r'/mnt/'+args.host+'/'+args.zone+'/' + rootDir +'/defaultVars/Vars/Connections'):
            for file in files:
                if file.endswith(".substvar"):
                    path = os.path.join(root, file)
                    data = {rootDir: {'EMS': {}}}
                    doc = xml.dom.minidom.parse(path)
                    for info in doc.getElementsByTagName("globalVariable"):
                         name = str(info.getElementsByTagName("name")[0].firstChild.data)
                         value = str(info.getElementsByTagName("value")[0].firstChild.data)
                         data[rootDir]['EMS'][name] = value                   
                    print(element, "\n", data, "\n")

[evgeh4ik]

Сергей Горностаев, не запускал пока, но вот вопрос: data = {rootDir: {'EMS': {}}}

EMS - это лишь одна из директорий, которую я хочу выводить в результате.
В той же папке есть:
EMS
VAR
INFO

То есть мне нужно что бы это была переменная, а не статичное имя папки.
Или я не верно понял код?)

[Сергей Горностаев]

evgeh4ik, подставляйте вместо строки нужную вам переменную, делов-то.

[evgeh4ik]

Сергей Горностаев, самый тупой вопрос. Прошу прощения заранее)) Но как мне указать, что следующая за Connections в случае

/mnt/'+args.host+'/'+args.zone+'/' + rootDir +'/defaultVars/Vars/Connections

папка - и есть моя переменная, которую нужно выводить? Понимаю, что это наверняка элементарно, но не могу найти инфы...
Возможно, нужно: data = {rootDir: {'Connections': {}}} ? т.к. вложенные в Connections папки меня как раз и интересуют...

[Сергей Горностаев]

evgeh4ik, плохое, но короткое решение - root.split('/')[-1]

[evgeh4ik]

Сергей Горностаев, А если говорить о хорошем решении?)

[Сергей Горностаев]

evgeh4ik, полностью переписать весь код. Он сильно кривой и в нём много лишнего. Его вполне реально вообще в одну строку уместить.

[evgeh4ik]

Сергей Горностаев, Использую root.split('/')[-1]

print(root.split('/')[-1], (json.dumps(data,indent = 4)))

Печатается только то, что было в последней директории.
То есть везде один и тот же путь:
Connections\Way4\TCP\Way4TS
Хотя фактически он долежн меняться.