test/test_base.py

   1 #!/usr/bin/env python
   2
   3 from htmltree import htmltree
   4 import sys
   5 import re
   6
   7 try:
   8     input = sys.argv[1]
   9 except IndexError:
  10     sys.exit("%s <input_html>" % sys.argv[0])
  11
  12 f = open(input, "r")
  13 html_text = f.read()
  14 f.close()
  15
  16 t = htmltree.HTMLTree()
  17 t.parse(html_text)
  18
  19 r = t.root()
  20
  21 titles = r.select_by_name2("head", "title")
  22 if len(titles) > 0:
  23     for title in titles:
  24         for item in title:
  25             print item
  26             print item.text()
  27
  28 rn = htmltree.HTMLRenderer()
  29 print rn.render(r)
  30
  31 t = r.get_element_by_id("htmltree")
  32 print t.inner_html()