Storage/bkmk_stflad.py

   1 """Bookmarks storage module - FLAD (Flat ASCII Database)
   2
   3 Special version for compatibility with old (version 1) bkmk2db.
   4
   5 This file is a part of Bookmarks database and Internet robot.
   6 """
   7
   8 __author__ = "Oleg Broytman <phd@phdru.name>"
   9 __copyright__ = "Copyright (C) 2000-2012 PhiloSoft Design"
  10 __license__ = "GNU GPL"
  11
  12 __all__ = ['storage_flad']
  13
  14
  15 import string
  16 from m_lib.flad import fladm
  17 from bkmk_objects import Folder, Bookmark, Ruler, Walker
  18
  19
  20 class storage_flad(Walker):
  21    filename = "bookmarks_db.flad"
  22
  23    def __init__(self):
  24       self.first_object = 1
  25
  26    def root_folder(self, f):
  27       header = string.replace(f.header, ".", ".\n")
  28       header = string.replace(header, "<", "\n<", 3)[1:]
  29       header_file = open("header", 'w')
  30       header_file.write(header + "\n")
  31       header_file.write('<H1>%s</H1>\n\n' % f.name)
  32       if f.comment: header_file.write('<DD>%s\n' % f.comment)
  33       header_file.close()
  34
  35    def start_folder(self, f, level):
  36       self.outfile.write("""
  37 Level: %d
  38 Folder: %s
  39 AddDate: %s
  40 Comment: %s
  41 LastModified: %s
  42 """ % (level, f.name, f.add_date, f.comment, f.last_modified or ''))
  43
  44    def bookmark(self, b, level):
  45       self.outfile.write("""
  46 Level: %d
  47 Title: %s
  48 URL: %s
  49 AddDate: %s
  50 LastVisit: %s
  51 LastModified: %s
  52 Keyword: %s
  53 Comment: %s
  54 IconURI: %s
  55 Icon: %s
  56 Charset: %s
  57 """ % (level+1, b.name, b.href, b.add_date, b.last_visit, b.last_modified,
  58          b.keyword, b.comment.replace('\n', "\\n"),
  59          b.icon_href or '', b.icon or '', b.charset or ''))
  60
  61    def ruler(self, r, level):
  62       self.outfile.write("\nLevel: %s\nRuler: YES\n" % (level+1))
  63
  64    def store(self, root_folder):
  65       self.outfile = open(self.filename, 'w')
  66       root_folder.walk_depth(self)
  67       self.outfile.close()
  68
  69    def unindent(self, old_level, new_level):
  70       while old_level > new_level:
  71          old_level = old_level - 1
  72          del self.folder_stack[-1]
  73
  74       if self.folder_stack:
  75          self.current_folder = self.folder_stack[-1]
  76       else:
  77          self.current_folder = None
  78
  79    def load(self):
  80       bookmarks_db = fladm.load_from_file(self.filename, fladm.check_record, ["Level"])
  81
  82       root_folder = Folder()
  83       self.folder_stack = [root_folder]
  84       self.current_folder = root_folder
  85
  86       header_file = open("header", 'r')
  87       header = header_file.read()
  88       header_file.close()
  89
  90       header = string.split(header, "\n")
  91       root_folder.header = string.join(header[:5], '')
  92       root_folder.name = header[5][4:-5]
  93       root_folder.comment = string.join(header[7:], '')[4:]
  94
  95       save_level = 0
  96       got_folder = 1 # Start as if we already have one folder
  97
  98       for record in bookmarks_db:
  99          level = int(record["Level"])
 100
 101          if level == save_level:
 102             pass
 103          elif level == save_level + 1:
 104             if not got_folder:
 105                raise ValueError, "indent without folder"
 106          elif level <= save_level - 1:
 107             self.unindent(save_level, level)
 108          else:
 109             raise ValueError, "new level (%d) too big; must be %d - %d" % (level, save_level-1, save_level+1)
 110
 111          save_level = level
 112          got_folder = record.has_key("Folder") # Test here to save got_folder for next loop
 113
 114          if record.has_key("URL"):
 115             comment = record["Comment"].replace("\\n", '\n')
 116             bookmark = Bookmark(record["URL"], record["AddDate"],
 117                record["LastVisit"], record["LastModified"],
 118                record["Keyword"], comment)
 119             bookmark.name = record["Title"]
 120             self.current_folder.append(bookmark)
 121
 122          elif record.has_key("Folder"):
 123             folder = Folder(record["AddDate"], record["Comment"], record["LastModified"])
 124             folder.name = record["Folder"]
 125             self.current_folder.append(folder)
 126             self.folder_stack.append(folder)
 127             self.current_folder = folder
 128
 129          elif record.has_key("Ruler"):
 130             self.current_folder.append(Ruler())
 131
 132          else:
 133             raise KeyError, "neither \"URL\" nor \"Folder\" nor \"Ruler\" in record " + str(record)
 134
 135       if save_level >= 0:
 136          self.unindent(save_level, 0)
 137       else:
 138          raise ValueError, "new level (%d) too little - must be >= 0" % save_level
 139
 140       return root_folder