X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;ds=sidebyside;f=Robots%2Fparse_html_beautifulsoup.py;h=e03dfce99faa55b980e21a75e38249790d50c0a7;hb=52092194ea42dcece57ed93c2a2875cd2907564e;hp=5f010ad698e894e63d86515fa2a0111ad926da4d;hpb=4ad8953212582076957a8fe471d627528765dffa;p=bookmarks_db.git

diff --git a/Robots/parse_html_beautifulsoup.py b/Robots/parse_html_beautifulsoup.py
index 5f010ad..e03dfce 100644
--- a/Robots/parse_html_beautifulsoup.py
+++ b/Robots/parse_html_beautifulsoup.py
@@ -1,23 +1,13 @@
 """
    HTML Parser using BeautifulSoup
 
-   Written by BroytMann. Copyright (C) 2007 PhiloSoft Design
+   Written by Broytman. Copyright (C) 2007-2010 PhiloSoft Design
 """
 
 import re
 from sgmllib import SGMLParser, SGMLParseError
-from HTMLParser import HTMLParser
 from BeautifulSoup import BeautifulSoup, CData
-
-
-class BSoupParser(HTMLParser):
-   def __init__(self, charset, meta, title, refresh, icon):
-      object.__init__(self)
-      self.charset = charset
-      self.meta_charset = meta
-      self.title = title
-      self.refresh = refresh
-      self.icon = icon
+from parse_html_util import HTMLParser
 
 
 # http://groups.google.com/group/beautifulsoup/browse_thread/thread/69093cb0d3a3cf63
@@ -39,7 +29,7 @@ class BadDeclParser(BeautifulSoup):
              except SGMLParseError:
                  # Could not parse the DOCTYPE declaration
                  # Try to just skip the actual declaration
-                 match = re.search(r'<!DOCTYPE([^>]*?)>', self.rawdata[i:], re.MULTILINE)
+                 match = re.search(r'<!DOCTYPE([^>]*?)>', self.rawdata[i:], re.MULTILINE|re.IGNORECASE)
                  if match:
                      toHandle = self.rawdata[i:match.end()]
                  else:
@@ -49,33 +39,92 @@ class BadDeclParser(BeautifulSoup):
          return j
 
 
-def parse_html(filename, charset=None):
+def _parse_html(filename, charset):
    infile = open(filename, 'r')
    try:
-      root = BadDeclParser(infile, fromEncoding=charset)
+      return BadDeclParser(infile, fromEncoding=charset)
    except TypeError:
       return None
-   infile.close()
+   finally:
+      infile.close()
+
+def parse_html(filename, charset=None, log=None):
+   root = _parse_html(filename, charset)
+   if root is None:
+      return None
 
    _charset = root.originalEncoding
-   try:
-      title = root.html.head.title.string.encode(_charset)
-   except AttributeError:
+   if _charset in ("ISO-8859-2", "windows-1252", "MacCyrillic"): # Replace default
+      _charset = DEFAULT_CHARSET
+      root = _parse_html(filename, _charset)
+      if root is None:
+         return None
+
+   html = root.html
+   if html is None:
+      html = root
+
+   head = html.head
+   if head is None:
+      head = html # Some sites put TITLE in HTML without HEAD
+
+   title = head.title
+   if (title is None) and (html is not head):
+      # Some sites put TITLE in HTML outside of HEAD
+      title = html.title
+
+   if title is None:
+      # Lookup TITLE in the root
+      title = root.title
+
+   if title is None:
       return None
 
-   meta = root.html.head.find(_find_refresh, recursive=False)
+   if title.string:
+      title = title.string
+   else:
+      parts = []
+      for part in title:
+         if not isinstance(part, basestring):
+            part = unicode(part)
+         parts.append(part.strip())
+      title = ''.join(parts)
+
+   meta = head.find(_find_contenttype, recursive=False)
+   if meta:
+      try:
+         meta_content = meta.get("content")
+         if meta_content:
+             __charset = meta_content.lower().split('charset=')[1].split(';')[0]
+         else:
+             __charset = False
+      except IndexError: # No charset in the META Content-Type
+         meta_charset = False
+      else:
+         meta_charset = _charset == __charset
+   else:
+      meta_charset = False
+
+   if _charset or meta_charset:
+      title = title.encode(_charset or meta_charset)
+
+   meta = head.find(_find_refresh, recursive=False)
    if meta:
       refresh = meta.get("content")
    else:
       refresh = None
 
-   meta = root.html.head.find(_find_icon, recursive=False)
+   meta = head.find(_find_icon, recursive=False)
    if meta:
       icon = meta.get("href")
    else:
       icon = None
 
-   return BSoupParser(_charset, _charset != charset, title, refresh, icon)
+   return HTMLParser(_charset, meta_charset, title, refresh, icon)
+
+def _find_contenttype(Tag):
+   return (Tag.name == "meta") and \
+      (Tag.get("http-equiv", '').lower() == "content-type")
 
 def _find_refresh(Tag):
    return (Tag.name == "meta") and \