set-title-list.py

   1 #! /usr/bin/env python
   2 """Run through the bookmarks database and set names to titles from an external file
   3
   4 This file is a part of Bookmarks database and Internet robot.
   5 """
   6
   7 from __future__ import print_function
   8 import sys
   9
  10
  11 __author__ = "Oleg Broytman <phd@phdru.name>"
  12 __copyright__ = "Copyright (C) 2003-2017 PhiloSoft Design"
  13 __license__ = "GNU GPL"
  14
  15
  16 def run():
  17    from getopt import getopt
  18    optlist, args = getopt(sys.argv[1:], "s")
  19
  20    report_stats = 1
  21
  22    for _opt, _arg in optlist:
  23       if _opt == '-s':
  24          report_stats = 0
  25    try:
  26       del _opt, _arg
  27    except NameError:
  28       pass
  29
  30    if report_stats:
  31       print("Broytman set-title-list, Copyright (C) 2003-2017 PhiloSoft Design")
  32
  33    if len(args) <> 1:
  34       sys.stderr.write("Usage: set-title-list [-s] title_list_file\n")
  35       sys.exit(1)
  36
  37    # Read the external file with titles and build a mapping (URL => title)
  38    titles_dict = {}
  39
  40    URL = None
  41    title = None
  42
  43    title_list_file = open(args[0], 'r')
  44    for line in title_list_file:
  45       line = line[:-1] # strip trailing newline
  46       if URL is None:
  47          URL = line
  48
  49       elif title is None:
  50          title = line
  51
  52       elif line: # the third line in every 3 lines must be empty
  53          raise ValueError("line is not empty for URL `%s', title `%s': line `%s'" % (URL, title, line))
  54
  55       else: # We've got 3 lines - add new entry to the mapping
  56          if titles_dict.has_key(URL):
  57             if title <> titles_dict[URL]:
  58                raise ValueError("titles are not identical for URL `%s': `%s' <> `%s'" % (URL, title, titles_dict[URL]))
  59
  60          else:
  61             titles_dict[URL] = title
  62
  63          # reset
  64          URL = None
  65          title = None
  66
  67    title_list_file.close()
  68
  69
  70    from storage import storage
  71    storage = storage()
  72
  73    if report_stats:
  74       sys.stdout.write("Loading %s: " % storage.filename)
  75       sys.stdout.flush()
  76
  77    root_folder = storage.load()
  78    from bkmk_objects import make_linear, break_tree
  79    make_linear(root_folder)
  80    objects = len(root_folder.linear)
  81
  82    if report_stats:
  83       print("Ok")
  84
  85
  86    # Run through the list of objects and check URLs/titles
  87    changed = 0
  88    for object_no in range(objects):
  89       object = root_folder.linear[object_no]
  90
  91       if object.isBookmark:
  92          URL = object.href
  93          if titles_dict.has_key(URL):
  94             name = titles_dict[URL]
  95             if object.name <> name:
  96                object.name = name
  97                changed += 1
  98
  99
 100    if changed and report_stats:
 101       sys.stdout.write("Saving %s: " % storage.filename)
 102       sys.stdout.flush()
 103
 104    if not changed and report_stats:
 105       sys.stdout.write("No need to save data\n")
 106       sys.stdout.flush()
 107
 108    if changed:
 109       break_tree(root_folder.linear)
 110       storage.store(root_folder)
 111
 112    if changed and report_stats:
 113       print("Ok")
 114       print(objects, "objects passed")
 115       print(changed, "objects changed")
 116
 117
 118 if __name__ == '__main__':
 119    run()