66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
import re
|
|
|
|
TAB_SIZE = 2
|
|
tab = 0
|
|
tabArray = 0
|
|
result = '{\n'
|
|
brackets = []
|
|
|
|
|
|
def append_to_result(string):
|
|
global result, tab, tabArray
|
|
result += f'{" " * (tabArray + tab + 2)}{string}'
|
|
|
|
|
|
with open('input.yaml', encoding='UTF-8') as inpt:
|
|
inpt = inpt.readlines()
|
|
for key, i in enumerate(inpt):
|
|
i = i.split(':')
|
|
|
|
# Обработка табов
|
|
deltaTab = len(re.findall(r'(?<!-) ', i[0]))
|
|
if tab != deltaTab:
|
|
if tab < deltaTab:
|
|
tab = deltaTab
|
|
else:
|
|
tab = deltaTab
|
|
result = result[:len(result) - 2] + '\n'
|
|
append_to_result(brackets.pop() + ',\n')
|
|
|
|
# Обработка массивов
|
|
if re.findall(r' *- .*', i[0]):
|
|
if tab == tabArray and ']' not in brackets:
|
|
append_to_result('[\n')
|
|
brackets.append(']')
|
|
tabArray += TAB_SIZE
|
|
i[0] = i[0].lstrip()[1:].strip()
|
|
append_to_result('{\n')
|
|
brackets.append('}')
|
|
tab += TAB_SIZE
|
|
|
|
# Обработка вложенных объектов
|
|
if i[1].strip() == '':
|
|
if re.findall(r' *- .*', inpt[key+1].split(':')[0]):
|
|
br = '[]'
|
|
else:
|
|
br = '{}'
|
|
append_to_result(f'"{i[0].strip()}": {br[0]}\n')
|
|
brackets.append(br[1])
|
|
else:
|
|
def shielding(s: str):
|
|
return s.replace("\\", "\\\\").replace("\"", "\\\"")
|
|
i = list(map(shielding, i))
|
|
append_to_result(f'"{i[0].strip()}": "{":".join(i[1:]).strip()}",\n')
|
|
# Закрытие файла
|
|
if brackets is not None:
|
|
for _ in range(len(brackets)):
|
|
tab -= TAB_SIZE
|
|
result = result[:len(result) - 2] + '\n'
|
|
append_to_result(brackets.pop() + ',\n')
|
|
result = result[:len(result) - 2] + '\n'
|
|
|
|
result += '}'
|
|
with open('output.json', 'w', encoding='UTF-8') as output:
|
|
output.write(result)
|
|
print(result)
|