Парсинг xml. Как обойти проблему с одной позицией в документе?

Question

[Максим]

Изучаю питон, в части парсинга xml и занесения данных в базу данных.
для парсинга использую библиотеку xmltodict, поскольку, она парсит файл и предоставляет его в виде словаря.

есть xml вида

<wb:Content>
  <wb:Position>
    <wb:Identity>1</wb:Identity>
    <wb:Product>
      <pref:Type>АП</pref:Type>
      <pref:FullName>Название</pref:FullName>
      <pref:AlcCode>0323103000001227178</pref:AlcCode>
      <pref:Capacity>0.500</pref:Capacity>
      <pref:UnitType>Packed</pref:UnitType>
      <pref:AlcVolume>4.000</pref:AlcVolume>
      <pref:ProductVCode>500</pref:ProductVCode>
      <pref:Producer>  ...      </pref:Producer>
    </wb:Product>
    <wb:Quantity>20.000</wb:Quantity>
    <wb:Price>57.19</wb:Price>
    <wb:Party>Партия №60074</wb:Party>
    <wb:FARegId>000000000000</wb:FARegId>
    <wb:InformF2>  ...  </wb:InformF2>
  </wb:Position>
</wb:Content>

проблема заключается в том, что элементов wb:Position может быть несколько.
учитывая специфику работы библиотеки если элемент wb:Position один, то обращение к его дочерним элементам будет происходить без использования вложенного списка. Например позиция всего одна:
xml['wb:Content']['wb:Position']['wb:Identity']

если их несколько то:

xml['wb:Content']['wb:Position'][3]['wb:Identity']

Заведомо не известно, сколько будет элементов wb:Position, если перебирать циклом, то можно поймать исключение, поскольку ключа 0 не существует.
Как проверять количество элементов wb:Position? или будет лучше использовать другой парсер?

Answer 1

[Алан Гибизов]

Ну можно тупо try и если exception, то значит вложен один элемент. Тупо, но если «на разовый скриптик» - пойдет наверно.

Comments

[Максим]

дело в том, что это не разовый "скриптик". к сожалению такой вариант меня не устраивает, поскольку нужно будет писать обработку для документа с одной позицией, и с несколькими, которая будет незначительно отличаться наличием/отсутствием итератора

[Алан Гибизов]

Максим, а ты проверяй тип значения словаря. Если тип - список, значит их там внутре несколько и надо перебирать.

import xmltodict

inxml = """<wb:Content>
  <wb:Position>
    <wb:Identity>1</wb:Identity>
    <wb:Product>
      <pref:Type>АП1</pref:Type>
     </wb:Product>
  </wb:Position>
  <wb:Position>
    <wb:Identity>2</wb:Identity>
    <wb:Product>
      <pref:Type>АП2-1</pref:Type>
      <pref:Type>АП2-2</pref:Type>
    </wb:Product>
  </wb:Position>
</wb:Content>"""

xml = xmltodict.parse(inxml)
print(type(xml['wb:Content']['wb:Position']))

[Максим]

Алан Гибизов, как?

[Алан Гибизов]

Максим,

import xmltodict

inxml = """<wb:Content>
  <wb:Position>
    <wb:Identity>1</wb:Identity>
    <wb:Product>
      <pref:Type>АП1</pref:Type>
     </wb:Product>
  </wb:Position>
  <wb:Position>
    <wb:Identity>2</wb:Identity>
    <wb:Product>
      <pref:Type>АП2-1</pref:Type>
      <pref:Type>АП2-2</pref:Type>
    </wb:Product>
  </wb:Position>
</wb:Content>"""

xml = xmltodict.parse(inxml)
print(type(xml['wb:Content']['wb:Position']))

[Максим]

Алан Гибизов, по прежнему осталась проблема дублирования кода. для одной позиции и с несколькими

[Алан Гибизов]

Максим, так приведи свой код.

[Алан Гибизов]

Максим, я имею ввиду - внятный кусок кода, чтоб видно было структуру программы.

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

Максим, ну смотри у тебя какая-то логика обработки этой позиции вынеси ее в отдельную функцию, класс метод не важно. И пусть она принимает только одну позицию и все делает.
И отдельно функцию которая проверят сколько там позиций, и если их несколько в цикле перебирает и вызывает эту функцию, если одна то соответственно только один раз.
ну и если тебе еще где-то эти данные нужны, можешь потом всегда упаковывать результат в список

Answer 2

[PavelMos]

Так сделать цикл и встроить обработку возможности отсутствия