src/FukuiNoNamari/barehtmlparser.py

   1 # Copyright (C) 2006 by Aiwota Programmer
   2 # aiwotaprog@tetteke.tk
   3 #
   4 # This program is free software; you can redistribute it and/or modify
   5 # it under the terms of the GNU General Public License as published by
   6 # the Free Software Foundation; either version 2 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program; if not, write to the Free Software
  16 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17
  18 from HTMLParserEx import HTMLParserEx
  19 import htmlentitydefs
  20
  21
  22 class BareHTMLParser(HTMLParserEx):
  23     """Parses html by the minimal necessity
  24
  25     to_out_func format is:
  26     def some_func(untied_data, is_bold, href):
  27     where untied_data is non markuped string
  28     and is_bold is whether untied_data is bold or not
  29     and href is url anchor if exists
  30
  31     strip spaces at the head and end of line, but first line's head is unable.
  32     """
  33
  34     def __init__(self, to_out_func):
  35         HTMLParserEx.__init__(self)
  36         self.to_out_func = to_out_func
  37         self.bold = False
  38         self.href = None
  39         self.buffer = ""
  40
  41     def reset_func(self, to_out_func):
  42         self.flush()
  43         self.to_out_func = to_out_func
  44
  45     def to_out(self, data):
  46         n = len(self.buffer)
  47         if n > 0 and self.buffer[n-1] == "\n":
  48             data = data.lstrip(" ")
  49         self.buffer = self.buffer + data
  50
  51     def flush(self):
  52         if self.buffer:
  53             self.to_out_func(self.buffer, self.bold, self.href)
  54             self.buffer = ""
  55
  56     def newline(self):
  57         self.buffer = self.buffer.rstrip(" ")
  58         self.flush()
  59         self.to_out("\n")
  60
  61     # override
  62     # flush after closing
  63     def close(self):
  64         HTMLParserEx.close(self)
  65         self.flush()
  66
  67     # override handle_*
  68
  69     def handle_starttag(self, tag, attr):
  70         if tag == "b":
  71             self.flush()
  72             self.bold = True
  73         elif tag == "br":
  74             self.newline()
  75         elif tag == "a":
  76             self.flush()
  77             for item in attr:
  78                 if item[0] == "href":
  79                     self.href = item[1]
  80
  81     def handle_endtag(self, tag):
  82         if tag == "b":
  83             self.flush()
  84             self.bold = False
  85         elif tag == "a":
  86             self.flush()
  87             self.href = None
  88
  89     def handle_data(self, data):
  90         self.to_out(data)
  91
  92     def handle_charref(self, ref):
  93         data = None
  94         try:
  95             data = unichr(int(ref))
  96         except:
  97             data = "&#"+ref+";"
  98         self.to_out(data)
  99
 100     def handle_entityref(self, name):
 101         if name in htmlentitydefs.name2codepoint:
 102             codepoint = htmlentitydefs.name2codepoint[name]
 103             self.to_out(unichr(codepoint))
 104         else:
 105             self.to_out("&"+name+";")