Как анализировать количество директорий в пути к файлу и выводить каждую на новой строке?

Question

[evgeh4ik]

Привет!

Есть такой код:

#!/usr/bin/env python36

import os
import xml.dom.minidom
import argparse
import re
import json

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_tag = info.getElementsByTagName("name")
		if len(name_tag) and name_tag[0].firstChild is not None:
			name = str(name_tag[0].firstChild.data)
			value_tag = info.getElementsByTagName("value")
			if len(value_tag) and value_tag[0].firstChild is not None:
				data[name] = str(value_tag[0].firstChild.data)
	print(element.split('/Connections/')[1], (json.dumps(data,indent = 4)))

Вот его результат выполнения:


Каким образом реализовать печать пути к файлу (каждой из директорий, выделенных красным) на новой строчке? И при этом исключить печать "defaultVars.substvar"?
Т.е. в формате (пример по первой строчке вывода):

{ EMS
 { DATA
   { Login }
   { Password }
   { URL }
  }
}

При этом, не везде 2 директории. Где-то 3, где-то 1.

Comments

[Влад Григорьев]

Итак если я правильно понял то вам нужно делать примерно так:

import json
# Ваш условный результат
tmp = {
    'EMS/Data/defaultVars.blabla': {
        'login': 'afsdfa',
        'users': 'afsdfa',
        'blala': 'afsdfa',
    },
    'EMS/Logs/defaultVars.blabla': {
        'login': 'afsdfa',
        'users': 'afsdfa',
        'blala': 'afsdfa',
    },
    'EMS/External/defaultVars.blabla': {
        'login': 'afsdfa',
        'users': 'afsdfa',
        'blala': 'afsdfa',
    },
    'Blabla/Data/defaultVars.blabla': {
        'login': 'afsdfa',
        'users': 'afsdfa',
        'blala': 'afsdfa',
    },
}

new_dict = {}


def append_value(modifier_dict: dict, key, value):
    if len(key) > 1:
        modifier_dict[key[0]] = append_value(modifier_dict.get(key[0], {}), key[1:], value)
    else:
        modifier_dict[key[0]] = value
    return modifier_dict


for k, v in tmp.items():
    append_value(new_dict, k.split('/'), v)
# Новый результат
print(json.dumps(new_dict, indent=4))

Результат

{
    "EMS": {
        "Data": {
            "defaultVars.blabla": {
                "login": "afsdfa",
                "users": "afsdfa",
                "blala": "afsdfa"
            }
        },
        "Logs": {
            "defaultVars.blabla": {
                "login": "afsdfa",
                "users": "afsdfa",
                "blala": "afsdfa"
            }
        },
        "External": {
            "defaultVars.blabla": {
                "login": "afsdfa",
                "users": "afsdfa",
                "blala": "afsdfa"
            }
        }
    },
    "Blabla": {
        "Data": {
            "defaultVars.blabla": {
                "login": "afsdfa",
                "users": "afsdfa",
                "blala": "afsdfa"
            }
        }
    }
}

Само собой правильнее это делать сразу пока заполняете data, туда и дополнительные проверки обрезки вставите.

Answer 1

[Владимир]

посмотрите этот пример. думаю то что Вам нужно

Comments

[evgeh4ik]

Смотрел, не подходит... Нужно из переменной, в которую входит список поочередно каждую строку перебирать, с условием что директория должна быть ниже Connections, и печатать каждую вложенную директорию на новой строке в json формате, желательно. Как реализовать - нет идей..