set-real_title.py

   1 #! /usr/bin/env python
   2 """Run through the bookmarks database and set name to real title
   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) 2002-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-real_title, Copyright (C) 2003-2017 PhiloSoft Design")
  32
  33    if args:
  34       sys.stderr.write("set-real_title: too many arguments\n")
  35       sys.stderr.write("Usage: set-real_title [-s]\n")
  36       sys.exit(1)
  37
  38    from storage import storage
  39    storage = storage()
  40
  41    if report_stats:
  42       sys.stdout.write("Loading %s: " % storage.filename)
  43       sys.stdout.flush()
  44
  45    root_folder = storage.load()
  46    from bkmk_objects import make_linear
  47    make_linear(root_folder)
  48    objects = len(root_folder.linear)
  49
  50    if report_stats:
  51       print("Ok")
  52
  53
  54    changed = 0
  55    for object_no in range(objects):
  56       object = root_folder.linear[object_no]
  57
  58       if object.isBookmark:
  59          if not hasattr(object, "real_title"):
  60             continue
  61
  62          real_title = object.real_title
  63          if not real_title:
  64             real_title = object.href
  65          if object.name <> real_title:
  66             object.name = real_title
  67             changed += 1
  68
  69
  70    if changed and report_stats:
  71       sys.stdout.write("Saving %s: " % storage.filename)
  72       sys.stdout.flush()
  73
  74    if not changed and report_stats:
  75       sys.stdout.write("No need to save data\n")
  76       sys.stdout.flush()
  77
  78    if changed:
  79       storage.store(root_folder)
  80
  81    if changed and report_stats:
  82       print("Ok")
  83       print(objects, "objects passed")
  84       print(changed, "objects changed")
  85
  86
  87 if __name__ == '__main__':
  88    run()