Iterating through xml in python through KeyError

Question

My Xml :

<books>
<book name="goodbook" cost="10" color="green"></book>
<book name="badbook" cost="1000" weight="100"></book>
<book name="avgbook" cost="99" weight="120"></book>
</books>

My python code :-

import xml.etree.ElementTree as ET
import sys
doc       = ET.parse("books.xml")
root      = doc.getroot() 
root_new  = ET.Element("books") 
for child in root:
       name                = child.attrib['name']
       cost                = child.attrib['cost']
       color               = child.attrib['color'] #KeyError
       weight              = child.attrib['weight'] #KeyError
       # create "book" here
       book    = ET.SubElement(root_new, "book") 
       book.set("name",name)               
       book.set("cost",cost) 
       book.set("color",color) 
       book.set("weight",weight)
tree = ET.ElementTree(root_new)
tree.write(sys.stdout)

What error am getting :-

python books.py 
Traceback (most recent call last):
  File "books.py", line 10, in <module>
    weight              = child.attrib['weight'] #KeyError
KeyError: 'weight'

weight and color are throughing keyerror, because while iterating through loop "color" and "weight" attribute not found in all line. i need my out put should be same as input xml :( . How can i skip this error and make it same as input xml . Thanks in advance.

light · Accepted Answer · 2012-11-06 04:48:13Z

8

for child in root:
    name                = child.attrib['name']
    cost                = child.attrib['cost']
    # create "book" here
    book    = ET.SubElement(root_new, "book") 
    book.set("name",name)               
    book.set("cost",cost) 
    if 'color' in child.attrib:
        color               = child.attrib['color']
        book.set("color",color) 
    if 'weight' in child.attrib:
        weight              = child.attrib['weight']
        book.set("weight",weight)

answered Nov 6, 2012 at 4:48

light

4212 silver badges2 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Sreeni Puthiyillam Over a year ago

Wow "light" you are really a light in program, super dude its working..With in 7 minutes i accept this answer. Thanks for your time:)

Collectives™ on Stack Overflow

Iterating through xml in python through KeyError

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related