sort_db.py

   1 #! /usr/bin/env python
   2 """Sort bookmarks DB according to a rule:
   3       -a - by AddDate
   4       -v - by LastVisit
   5       -m - by LastModified
   6       -z - by Size
   7       -t - by LastTested
   8    default is -m
   9       -r - reverse the sort order
  10
  11 This file is a part of Bookmarks database and Internet robot.
  12
  13 """
  14
  15 __author__ = "Oleg Broytman <phd@phdru.name>"
  16 __copyright__ = "Copyright (C) 2000-2014 PhiloSoft Design"
  17 __license__ = "GNU GPL"
  18
  19 import sys
  20
  21
  22 class SortBy(object):
  23    def __init__(self, sort_by):
  24       self.sort_by = sort_by
  25
  26    def __call__(self, o1, o2):
  27       try:
  28          attr1 = int(getattr(o1, self.sort_by))
  29       except (AttributeError, TypeError, ValueError):
  30          return 1
  31
  32       try:
  33          attr2 = int(getattr(o2, self.sort_by))
  34       except (AttributeError, TypeError, ValueError):
  35          return -1
  36
  37       return cmp(attr1, attr2)
  38
  39
  40 def walk_linear(linear, walker):
  41    for object in linear:
  42       if object.isBookmark:
  43          walker.bookmark(object, 0)
  44
  45
  46 def run():
  47    from getopt import getopt
  48    optlist, args = getopt(sys.argv[1:], "avmztrs")
  49
  50    sort_by = "last_modified"
  51    reverse = 0
  52    report_stats = 1
  53
  54    for _opt, _arg in optlist:
  55       if _opt == '-a':
  56          sort_by = "add_date"
  57       elif _opt == '-v':
  58          sort_by = "last_visit"
  59       elif _opt == '-m':
  60          sort_by = "last_modified"
  61       elif _opt == '-z':
  62          sort_by = "size"
  63       elif _opt == '-t':
  64          sort_by = "last_tested"
  65       elif _opt == '-r':
  66          reverse = 1
  67       elif _opt == '-s':
  68          report_stats = 0
  69    try:
  70       del _opt, _arg
  71    except NameError:
  72       pass
  73
  74    from storage import storage
  75    storage = storage()
  76
  77    if report_stats:
  78       sys.stdout.write("Loading %s: " % storage.filename)
  79       sys.stdout.flush()
  80
  81    root_folder = storage.load()
  82
  83    if report_stats:
  84       print "Ok"
  85       sys.stdout.write("Sorting (by %s): " % sort_by)
  86       sys.stdout.flush()
  87
  88    from bkmk_objects import make_linear
  89    make_linear(root_folder)
  90
  91    linear = root_folder.linear
  92    del linear[0] # exclude root folder from sorting
  93
  94    by = SortBy(sort_by)
  95    linear.sort(by)
  96
  97    from writers import writer
  98    output_filename = "%s-sorted_by-%s" % (writer.filename, sort_by)
  99
 100    if reverse:
 101       linear.reverse()
 102       output_filename = output_filename + "-reverse"
 103
 104    if report_stats:
 105       print "done"
 106       sys.stdout.write("Writing %s: " % output_filename)
 107       sys.stdout.flush()
 108
 109    outfile = open(output_filename, 'w')
 110    writer = writer(outfile)
 111    writer.root_folder(root_folder)
 112    walk_linear(linear, writer)
 113    outfile.close()
 114
 115    if report_stats:
 116       print "Ok"
 117
 118
 119 if __name__ == '__main__':
 120    run()