Add a few TODO items

[bookmarks_db.git] / bkmk_objects.py

diff --git a/bkmk_objects.py b/bkmk_objects.py
index fabf0ed5be81b6601ef3801e66d0fb026ccb09df..23fb816ee180100100722ee2b7ac9818cf63390d 100644 (file)
--- a/bkmk_objects.py
+++ b/bkmk_objects.py
@@ -1,18 +1,29 @@
-"""
-   Objects to represent bookmarks.html structure
+"""Objects to represent bookmarks.html structure
+
+This file is a part of Bookmarks database and Internet robot.
 
-   Written by BroytMann, Mar 2000 - Sep 2007. Copyright (C) 2000-2007 PhiloSoft Design
 """
 
+__author__ = "Oleg Broytman <phd@phdru.name>"
+__copyright__ = "Copyright (C) 2000-2012 PhiloSoft Design"
+__license__ = "GNU GPL"
+
+__all__ = ['Folder', 'Bookmark', 'Ruler', 'Walker', 'Writer', 'Robot',
+    'InverseLinker', 'Linear', 'make_linear', 'make_tree', 'break_tree',
+    'quote_title', 'unquote_title',
+]
+
 
-from UserList import UserList
+import os, urllib
 
-class Folder(UserList):
+BKMK_FORMAT = os.environ.get("BKMK_FORMAT", "MOZILLA")
+
+class Folder(list):
    isFolder = 1
    isBookmark = 0
 
-   def __init__(self, add_date = None, comment = '', last_modified=None):
-      UserList.__init__(self)
+   def __init__(self, add_date=None, comment='', last_modified=None):
+      super(Folder, self).__init__()
       self.comment = comment
       self.add_date = add_date
       self.last_modified = last_modified
@@ -27,7 +38,7 @@ class Folder(UserList):
             walker.start_folder(self, level)
 
       if not prune:
-         for object in self.data:
+         for object in self:
             if object.isFolder:
                object.walk_depth(walker, level+1)
             elif object.isBookmark:
@@ -43,13 +54,43 @@ class Bookmark:
    isBookmark = 1
 
    def __init__(self, href, add_date, last_visit=None, last_modified=None,
-         keyword=None, comment = ''):
-      self.comment = comment
+         keyword=None, comment='', icon_href=None, icon=None,
+         charset=None, parser_charset=None):
+      protocol, request = urllib.splittype(href)
+      user, password, port = None, None, None
+      host, path = urllib.splithost(request)
+      if host:
+         user, host = urllib.splituser(host)
+         if user:
+            user, password = urllib.splitpasswd(user)
+         host, port = urllib.splitport(host)
+         if port: port = int(port)
+
+      if protocol == 'place':
+         href = protocol + ":"
+      else:
+         href = protocol + "://"
+      if user:
+         href += urllib.quote(user)
+         if password:
+            href += ':' + urllib.quote(password)
+         href += '@'
+      if host:
+         href += host.decode(parser_charset or 'utf-8').encode('idna')
+         if port:
+            href += ':%d' % port
+      if path:
+         href += path
+
       self.href = href
       self.add_date = add_date
       self.last_visit = last_visit
       self.last_modified = last_modified
       self.keyword = keyword
+      self.comment = comment
+      self.icon_href = icon_href
+      self.icon = icon
+      self.charset = charset
 
 
 class Ruler:
@@ -95,8 +136,7 @@ class Writer(Walker):
 
 
 class Robot:
-   def __init__(self, tempfname, log):
-      self.tempfname = tempfname
+   def __init__(self, log):
       self.log = log
 
    def stop(self):
@@ -159,3 +199,22 @@ def make_tree(linear):
       object.parent.append(object)
 
    return root_folder
+
+def break_tree(linear):
+   del linear[0]
+
+   for object in linear:
+      del object.parent
+
+
+def quote_title(title):
+   if BKMK_FORMAT == "MOZILLA":
+      title = title.replace("'", "&#39;")
+   return title
+
+def unquote_title(title):
+   if BKMK_FORMAT == "MOZILLA":
+      from HTMLParser import HTMLParser
+      title = HTMLParser().unescape(title.replace("&amp;", '&').decode('utf-8'))
+      title = title.encode('utf-8').replace("&#39;", "'")
+   return title