Как преобразовать многострочный INI-файл в однострочный INI-файл в Python? - PullRequest
Новая
       41

Как преобразовать многострочный INI-файл в однострочный INI-файл в Python?

1 голос
/ 06 апреля 2020

У меня есть INI-файл, отформатированный так:

enter image description here

Но мне нужно, чтобы он выглядел так:

enter image description here

Что было бы самым простым решением для написания такого конвертера? Я пытался сделать это в Python, но это не сработало, как ожидалось. Мой код ниже. 

def fix_INI_file(in_INI_filepath, out_INI_filepath):
count_lines = len(open( in_INI_filepath).readlines() )
print("Line count: " + str(count_lines))

in_INI_file = open(in_INI_filepath, 'rt')

out_arr = []
temp_arr = []
line_flag = 0
for i in range(count_lines):
    line = in_INI_file.readline()
    print (i)

    if line == '':
        break

    if (line.startswith("[") and "]" in line)   or   ("REF:" in line)    or   (line == "\n"):
        out_arr.append(line)
    else:
        temp_str = ""
        line2 = ""
        temp_str = line.strip("\n")

        wh_counter = 0
        while 1:             
            wh_counter += 1
            line = in_INI_file.readline()
            if (line.startswith("[") and "]" in line)   or   ("REF:" in line)    or   (line == "\n"):
                line2 += line
                break
            count_lines -= 1
            temp_str += line.strip("\n") + " ; "    
        temp_str += "\n"
        out_arr.append(temp_str)
        out_arr.append(line2 )


out_INI_file = open(out_INI_filepath, 'wt+')  
strr_blob = ""
for strr in out_arr:
    strr_blob += strr
out_INI_file.write(strr_blob)


out_INI_file.close()
in_INI_file.close()

1 Ответ

1 голос
/ 06 апреля 2020

К счастью, есть гораздо более простой способ справиться с этим, чем анализировать текст вручную. Встроенный модуль configparser поддерживает ключи без значений через аргумент конструктора allow_no_values. 

import configparser


read_config = configparser.ConfigParser(allow_no_value=True)
read_config.read_string('''
[First section]
s1value1
s1value2

[Second section]
s2value1
s2value2
''')

write_config = configparser.ConfigParser(allow_no_value=True)

for section_name in read_config.sections():
    write_config[section_name] = {';'.join(read_config[section_name]): None}

with open('/tmp/test.ini', 'w') as outfile:
    write_config.write(outfile)

Пока я не сразу вижу способ чтобы использовать тот же объект ConfigParser для чтения и записи (он поддерживает значения по умолчанию для исходных ключей), использование второго объекта в качестве модуля записи должно дать то, что вы ищете.

Вывод из приведенного выше примера : 

[First section]
s1value1;s1value2

[Second section]
s2value1;s2value2
...