X-Git-Url: https://git.phdru.name/?p=bookmarks_db.git;a=blobdiff_plain;f=bkmk_objects.py;h=dce981190c217b7602bcd600b53055dd47c13488;hp=a9567e22134fd70b9adcefe788c2b2bb3a45746f;hb=96e39d55c791326368ff14d538850e14ba5a2c97;hpb=d9f2f4316537aea2cf87b2f77bf7d78eb54978a4

diff --git a/bkmk_objects.py b/bkmk_objects.py
index a9567e2..dce9811 100644
--- a/bkmk_objects.py
+++ b/bkmk_objects.py
@@ -1,12 +1,22 @@
-"""
-   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 Oleg Broytman. Copyright (C) 2000-2010 PhiloSoft Design.
 """
 
-import os, cgi
-BKMK_FORMAT = os.environ.get("BKMK_FORMAT", "MOZILLA")
+__author__ = "Oleg Broytman <phd@phdru.name>"
+__copyright__ = "Copyright (C) 2000-2014 PhiloSoft Design"
+__license__ = "GNU GPL"
 
+__all__ = ['Folder', 'Bookmark', 'Ruler', 'Walker', 'Writer', 'Robot',
+    'InverseLinker', 'Linear', 'make_linear', 'make_tree', 'break_tree',
+    'quote_title', 'unquote_title', 'parse_params', 'set_params',
+]
+
+
+import os, urllib
+
+BKMK_FORMAT = os.environ.get("BKMK_FORMAT", "MOZILLA")
 
 class Folder(list):
    isFolder = 1
@@ -39,12 +49,39 @@ class Folder(list):
          walker.end_folder(self, level)
 
 
-class Bookmark:
+class Bookmark(object):
    isFolder = 0
    isBookmark = 1
 
    def __init__(self, href, add_date, last_visit=None, last_modified=None,
-         keyword=None, comment='', icon_href=None, icon=None, charset=None):
+         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
@@ -56,12 +93,12 @@ class Bookmark:
       self.charset = charset
 
 
-class Ruler:
+class Ruler(object):
    isFolder = 0
    isBookmark = 0
 
 
-class Walker:
+class Walker(object):
    """
       Interface class. Any instance that will be passed to Folder.walk_depth
       may be derived from this class. It is not mandatory - unlike Java
@@ -98,9 +135,8 @@ class Writer(Walker):
       return self.prune == folder.name
 
 
-class Robot:
-   def __init__(self, tempfname, log):
-      self.tempfname = tempfname
+class Robot(object):
+   def __init__(self, log):
       self.log = log
 
    def stop(self):
@@ -179,6 +215,22 @@ def quote_title(title):
 def unquote_title(title):
    if BKMK_FORMAT == "MOZILLA":
       from HTMLParser import HTMLParser
-      title = HTMLParser().unescape(title.replace("&amp;", '&'))
-      title = title.replace("&#39;", "'")
+      title = HTMLParser().unescape(title.replace("&amp;", '&').decode('utf-8'))
+      title = title.encode('utf-8').replace("&#39;", "'")
    return title
+
+
+def parse_params(param_str):
+    params = param_str.split(':')
+    main_param = params.pop(0)
+    param_list = {}
+    for param in params:
+        key, value = param.split('=', 1)
+        param_list[key] = value
+    return main_param, param_list
+
+def set_params(obj, params):
+    if hasattr(params, "items"):
+        params = params.items()
+    for key, value in params:
+        setattr(obj, key, value)