Оптимизация несложного кода для автозаполнения XML

import xml.etree.ElementTree as ET
 
## Открываем и сохраняем наш 415-ый отчёт
tree = ET.parse('/home/runner/FruitfulFrugalObject/report.xml')
root = tree.getroot()
## Выводим номер накладной в консоль для удобства
print(root[0][3].text)
## Находим все нужные элементы(содержащие серию и цены)
unions = root[0][8]
## Считаем их количество
positions = len(unions.getchildren())
## Создаём элементы для хранения серий и цен
all_series = []
series_dict = {}
prices_dict = {}
 
 
 
## Открываем и сохраняем словарь с ценами
with open("/home/runner/FruitfulFrugalObject/prices.txt", 'r', encoding='UTF-8') as f:
  for line in f.readlines():
    prices_dict[line.split(":")[0]] = line.split(":")[1].strip('\n')
 
## Заполняем список серий
for i in range(positions):
    series = unions[i][0][1][1].text
    all_series.append(series)
 
## Выводим количество уникальных серий
all_series = list(dict.fromkeys(all_series))
print("Number of unique batches:", len(all_series))
 
## Проверяем есть ли в целом серии со скидкой
disc = input('Any discount? (y/n)')
if disc == 'Y' or disc == 'y':
## Если есть серии со скидкой
  for batch in all_series:
    print(batch)
    ## Проверяем каждую серию по отдельности, применялась ли к ней скидка
    ans = input('Discount? (y/n)')
    ## Если нет - то пробуем найти в словаре с ценами, если и там нет - вводим в ручную и она сохраянется в словаре
    if ans == 'n' or ans == 'N':
      if batch not in prices_dict.keys():
        price = float(
            input(f"Cost (w/o VAT) of {batch}: ").replace(",",
                                                          ".").replace(" ", ""))
        series_dict[batch] = price
        prices_dict[batch] = price
      else:
        series_dict[batch] = float(prices_dict[batch])
    ## Если эта серия со скидкой, просто вводим цену из excel
    else:
      price = float(
            input(f"Cost (w/o VAT) of {batch}: ").replace(",",
                                                          ".").replace(" ", ""))
      series_dict[batch] = price
      
## Если скидко нет вообще, то просто сравниваем с нашим словарём цен и вписываем те цены, которых там нет
else: 
  for batch in all_series:  
    if batch not in prices_dict.keys():
      print(batch)
      price = float(
          input(f"Cost (w/o VAT) of {batch}: ").replace(",",
                                                        ".").replace(" ", ""))
      series_dict[batch] = price
      prices_dict[batch] = price
    else:
      series_dict[batch] = float(prices_dict[batch])
      
## Выводим получившийся результат (серия - цена) для проверки
print(f"Check: {series_dict}")
 
## Вставляем цены в нужные места 415-го отчёта
for i in range(positions):
    series = unions[i][0][1][1].text
    if series in series_dict.keys():
      price = series_dict[series]
      unions[i][0][1][2].text = str(round(price * 1.1, 2))
      unions[i][1].text = str(round(price * 1.1, 2))
      unions[i][0][1][3].text = str(round(price * .1, 2))
      unions[i][2].text = str(round(price * .1, 2))
    else:
        print(f'A series not included in the dictionary was found: {series}')
 
## Сохраняем дополненный словарь с ценами обратно
with open("/home/runner/FruitfulFrugalObject/prices.txt", 'w', encoding='UTF-8') as w:
  for k,v in prices_dict.items():
    w.write(f"{k}:{v}\n")
 
## Сохраняем получившийся 415 отчёт в виде xml файла
tree.write(f"RU415_{root[0][3].text}.xml")