python-playground/xml_reader.py at master · caioboffo/python-playground · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#!/usr/bin/python

import xml.sax


class MovieHandler(xml.sax.ContentHandler):
    def __init__(self):
       self.CurrentData = ""
       self.type = ""
       self.format = ""
       self.year = ""
       self.rating = ""
       self.stars = ""
       self.description = ""

    # Call when an element starts
    def startElement(self, tag, attributes):
       self.CurrentData = tag
       if tag == "movie":
          print "*****Movie*****"
          title         = attributes["title"]
          print "Title:", title

    # Call when an elements ends
    def endElement(self, tag):
       if self.CurrentData == "type":
          print "Type:", self.type
       elif self.CurrentData == "format":
          print "Format:", self.format
       elif self.CurrentData == "year":
          print "Year:", self.year
       elif self.CurrentData == "rating":

          print "Rating:", self.rating
       elif self.CurrentData == "stars":
          print "Stars:", self.stars
       elif self.CurrentData == "description":
          print "Description:", self.description
          self.CurrentData = ""

    # Call when a character is read
    def characters(self, content):
       if self.CurrentData == "type":
          self.type   = content
       elif self.CurrentData == "format":
          self.format = content
       elif self.CurrentData == "year":
          self.year   = content
       elif self.CurrentData == "rating":
          self.rating = content
       elif self.CurrentData == "stars":
          self.stars  = content
       elif self.CurrentData == "description":
          self.description = content

if ( __name__ == "__main__"):

   # create an XMLReader
   parser = xml.sax.make_parser()
   # turn off namepsaces
   parser.setFeature(xml.sax.handler.feature_namespaces, 0)

   # override the default ContextHandler
   Handler = MovieHandler()
   parser.setContentHandler( Handler )

   parser.parse("movies.xml")