X-Git-Url: https://git.phdru.name/?p=bookmarks_db.git;a=blobdiff_plain;f=db2bkmk.py;h=e0ac1c1bcb6b76c6247f7680421a2e66765b34a4;hp=28ebc76959e5fb6d470e4a1d3fb92be1fee4a7bf;hb=c88cb7a75e7caf1d67466cfa107981d95115fa0c;hpb=a04eaa0346e8aa5ad86a195f8f4d36487ebfe09c diff --git a/db2bkmk.py b/db2bkmk.py index 28ebc76..e0ac1c1 100755 --- a/db2bkmk.py +++ b/db2bkmk.py @@ -12,97 +12,97 @@ import sys def run(): - from getopt import getopt - optlist, args = getopt(sys.argv[1:], "sp:o:t:r") + from getopt import getopt + optlist, args = getopt(sys.argv[1:], "sp:o:t:r") - report_stats = 1 - prune = None + report_stats = 1 + prune = None - from writers import writer - output_filename = writer.filename + from writers import writer + output_filename = writer.filename - transl = 0 - transl_name = "" # dictionary translation; default is no translation + transl = 0 + transl_name = "" # dictionary translation; default is no translation - for _opt, _arg in optlist: - if _opt == '-s': - report_stats = 0 - if _opt == '-p': - prune = _arg - if _opt == '-o': - output_filename = _arg - if _opt == '-t': - transl = 1 - transl_name = _arg - if _opt == '-r': - transl = 2 - try: - del _opt, _arg - except NameError: - pass + for _opt, _arg in optlist: + if _opt == '-s': + report_stats = 0 + if _opt == '-p': + prune = _arg + if _opt == '-o': + output_filename = _arg + if _opt == '-t': + transl = 1 + transl_name = _arg + if _opt == '-r': + transl = 2 + try: + del _opt, _arg + except NameError: + pass - if args: - sys.stderr.write("db2bkmk: too many arguments\n") - sys.stderr.write("Usage: db2bkmk [-s] [-p prune_folder] [-o filename] [-t trans] [-r]\n") - sys.exit(1) + if args: + sys.stderr.write("db2bkmk: too many arguments\n") + sys.stderr.write("Usage: db2bkmk [-s] [-p prune_folder] [-o filename] [-t trans] [-r]\n") + sys.exit(1) - from storage import storage - storage = storage() + from storage import storage + storage = storage() - if report_stats: - sys.stdout.write("Loading %s: " % storage.filename) - sys.stdout.flush() + if report_stats: + sys.stdout.write("Loading %s: " % storage.filename) + sys.stdout.flush() - root_folder = storage.load() + root_folder = storage.load() - if report_stats: - print("Ok") - sys.stdout.write("Writing %s: " % output_filename) - sys.stdout.flush() + if report_stats: + print("Ok") + sys.stdout.write("Writing %s: " % output_filename) + sys.stdout.flush() - if transl: - new_ext = str(transl) - transl_d = {} + if transl: + new_ext = str(transl) + transl_d = {} - from m_lib.flad import fladm - transl_db = fladm.load_from_file(transl_name, fladm.check_record, ["URL1", "URL2"], [""]) - # This prevents any other key to appear in transl_db ^ + from m_lib.flad import fladm + transl_db = fladm.load_from_file(transl_name, fladm.check_record, ["URL1", "URL2"], [""]) + # This prevents any other key to appear in transl_db ^ - # Generate translation dictionary (hash table) - if transl == 1: - for record in transl_db: - transl_d[record["URL1"]] = record["URL2"] - elif transl == 2: - for record in transl_db: - transl_d[record["URL2"]] = record["URL1"] - else: - raise ValueError("transl (%d) must be 1 or 2" % transl) + # Generate translation dictionary (hash table) + if transl == 1: + for record in transl_db: + transl_d[record["URL1"]] = record["URL2"] + elif transl == 2: + for record in transl_db: + transl_d[record["URL2"]] = record["URL1"] + else: + raise ValueError("transl (%d) must be 1 or 2" % transl) - del transl_db # Save few bytes of memory + del transl_db # Save few bytes of memory - from bkmk_objects import Walker - class Transl(Walker): - def __init__(self, transl_d): - self.transl_d = transl_d + from bkmk_objects import Walker + class Transl(Walker): + def __init__(self, transl_d): + self.transl_d = transl_d - def bookmark(self, b, level): - href = b.href - transl_d = self.transl_d + def bookmark(self, b, level): + href = b.href + transl_d = self.transl_d - if transl_d.has_key(href): - b.href = transl_d[href] + if transl_d.has_key(href): + b.href = transl_d[href] - root_folder.walk_depth(Transl(transl_d)) + root_folder.walk_depth(Transl(transl_d)) - outfile = open(output_filename, 'w') - root_folder.walk_depth(writer(outfile, prune)) - outfile.close() + outfile = open(output_filename, 'w') + root_folder.walk_depth(writer(outfile, prune)) + outfile.close() - if report_stats: - print("Ok") + if report_stats: + print("Ok") if __name__ == '__main__': - run() + run()