X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=Robots%2Fbkmk_rsimple.py;h=ba302943d175dd5e969adcf82cf564e0f2cb644b;hb=1d46a18bfe213e25401908cc3c31c5ac66008572;hp=09b1fbdb11880e96eef0a3ffbbb934248b9e5504;hpb=dcd9c42a34f5ce49b49c08f35c5629f468641b64;p=bookmarks_db.git diff --git a/Robots/bkmk_rsimple.py b/Robots/bkmk_rsimple.py index 09b1fbd..ba30294 100644 --- a/Robots/bkmk_rsimple.py +++ b/Robots/bkmk_rsimple.py @@ -1,7 +1,7 @@ """ Simple, strightforward robot - Written by Oleg BroytMann. Copyright (C) 2000-2008 PhiloSoft Design. + Written by Oleg Broytman. Copyright (C) 2000-2010 PhiloSoft Design. """ import sys, os @@ -158,6 +158,7 @@ class robot_simple(Robot): if headers: try: content_type = headers["Content-Type"] + self.log(" Content-Type: %s" % content_type) try: # extract charset from "text/html; foo; charset=UTF-8, bar; baz;" content_type, charset = content_type.split(';', 1) @@ -167,7 +168,13 @@ class robot_simple(Robot): except (ValueError, IndexError): charset = None self.log(" no charset in Content-Type header") - if content_type in ("text/html", "application/xhtml+xml"): + for ctype in ("text/html", "application/xhtml+xml"): + if content_type.startswith(ctype): + html = True + break + else: + html = False + if html: parser = parse_html(fname, charset, self.log) bookmark.real_title = parser.title if parser.refresh: @@ -195,6 +202,7 @@ class robot_simple(Robot): self.log(" looking for icon at: %s" % icon) if icon in icons: if icons[icon]: + bookmark.icon_href = icon content_type, bookmark.icon = icons[icon] self.log(" cached icon: %s" % content_type) else: @@ -223,9 +231,10 @@ class robot_simple(Robot): content_type = headers["Content-Type"] if content_type.startswith("image/"): icon_file = open(fname, "rb") - icon = icon_file.read() + icon_data = icon_file.read() icon_file.close() - bookmark.icon = "data:%s;base64,%s" % (content_type, b64encode(icon)) + bookmark.icon_href = icon + bookmark.icon = "data:%s;base64,%s" % (content_type, b64encode(icon_data)) self.log(" got icon : %s" % content_type) icons[icon] = (content_type, bookmark.icon) else: @@ -233,28 +242,34 @@ class robot_simple(Robot): icons[icon] = None os.remove(fname) - except KeyError: - pass + except KeyError, key: + self.log(" no header: %s" % key) except IOError, msg: if (msg[0] == "http error") and (msg[1] == -1): bookmark.no_error = "The server did not return any header - it is not an error, actually" + self.log(bookmark.no_error) else: bookmark.error = get_error(msg) + self.log(bookmark.error) except EOFError: bookmark.error = "Unexpected EOF (FTP server closed connection)" + self.log(bookmark.error) except RedirectException, msg: bookmark.moved = str(msg) + self.log(bookmark.moved) except KeyboardInterrupt: + self.log("Keyboard interrupt (^C)") return 0 except: import traceback traceback.print_exc() bookmark.error = "Exception!" + self.log(bookmark.error) finally: self.finish_check_url(bookmark)